把 NumPy 风格的 Python 代码自动微分、向量化并 JIT 编译到 GPU/TPU,写起来像 NumPy,跑起来像高性能计算。
面向人类友好的深度学习 API,用 Python 编写,可跑在 TensorFlow、JAX、PyTorch 上,帮开发者快速搭建和训练神经网络。
每周更新的机器学习 Python 库排行榜,按类别和评分整理优质库,帮你快速找到 AutoML、深度学习、数据可视化等方向的趁手工具。
把 PyTorch、TensorFlow、JAX、NumPy 等框架的机器学习代码互相转换,省去为不同框架重写模型的麻烦。
用一套可读的 Einstein 记号统一 PyTorch、JAX、TensorFlow 等框架的张量操作,让 reshape、转置、rearrange 这类代码更清晰、更不容易出错。
从公开研究文献出发,用第一性原理复现 Claude Mythos 架构的理论实现,供研究者拆解其注意力机制与模型设计思路。
面向 AI 开发者的实验追踪与模型管理平台,记录训练指标、超参数和数据集版本,支持从实验到生产的全流程协作。
提供大量可直接用于 TensorFlow、JAX 等框架的数据集集合,省去自己下载、解析和预处理数据的麻烦,统一接口加载即可训练模型。
基于 JAX 的计算机视觉研究库,提供注意力机制、Transformer 和 Vision Transformer 等模型实现,方便快速搭建和复现视觉研究实验。
基于 MLIR 的可重定向机器学习编译器与运行时工具包,把 PyTorch、TensorFlow、JAX、ONNX 等模型编译到 CPU、CUDA、ROCm、Vulkan 等多种硬件上运行。
Atom 编辑器里的 Markdown 实时预览插件,支持 KaTeX/MathJax 公式、Mermaid/PlantUML 图表和 Pandoc 导出,写技术文档时预览效果一步到位。
在 Kubernetes 上跑分布式模型训练和大模型微调的框架,把多机多卡任务编排成 K8s 原生作业,省去自己搭训练集群的麻烦。
Kubernetes 上的自动机器学习平台,支持超参数调优和神经网络架构搜索,可对接 PyTorch、JAX 等训练框架。项目已弃用,官方建议迁移到 OptimizationJob 和 Kubeflow Trainer。
在免费 Kaggle TPU v5e-8 上跑 GLM-5.3-Flash 320B MoE 和 Qwen3.8-27B 等前沿大模型,自研 JAX 推理引擎支持 262k 长上下文和前缀缓存,可接入 Claude Code 等编码代理。
轻量级车牌文字识别 OCR 模型,支持 Keras、JAX 和 ONNX Runtime,可快速集成到车牌识别流水线中。
汇总 Transformer 自然语言处理模型的论文、视频、博客、官方代码和 Colab 笔记本,方便快速查找和上手各类预训练模型。
高性能机器学习模型服务框架,支持动态批处理和 CPU/GPU 流水线,把单机算力充分利用起来,适合部署 CV、LLM 等模型。
面向 XLA 设备(先支持 TPU,后续扩展 GPU)的大模型推理引擎,主打高吞吐和低显存占用,适合在 JAX 生态里跑 Llama、Gemma 等模型。
从零实现的大模型蒸馏与推理服务引擎,包含自定义 Triton/CUDA 算子、FSDP 蒸馏、分页 KV 连续批处理、投机解码和 Rust 网关,适合研究推理加速与可解释性。
一套完全开源的强文本到图像扩散模型训练配方,提供 JAX 和 PyTorch 实现,方便复现和二次开发。
一本博客风格的短教材,讲怎么在 TPU 上扩展大模型训练与推理,用 roofline 分析帮你算清算力、显存和通信的瓶颈。
Bug fixes Built C++-generated intrinsic LLVM IR on Windows with clang-cl instead of emitting empty bitcode, enabling vectorized CPU intrinsics on Windows (https
New features Added jax.scipy.special.erfcinv for the inverse complementary error function. Added jax.lax.polynomial for polynomial evaluation. The polynomial primitive is
New features Added jax.numpy.minmax (and jnp.minmax ), which returns both the minimum and maximum of an array, matching NumPy 2.3+ ( 40089). Added jax.lax.log2 and primit
…es Added an error check for trying to deserialize JAX exports that are older than the backwards compatibility window. Without this check the deserialization of expired ar
…g custom derivative rules with the experimental hijax API ( hijax-custom-derivatives ), along with jax.experimental.hijax helpers for deriving VJPHiPrimitive autodiff rul
… variants, support for google's new ML framework [JAX](https://github.com/google/jax), hyperparameter tuning utilities, and more. CleanRL has come a long way making high-