01最新发版
-
可组合变换的数值计算
-
JAX 与 OpenAI Triton 的集成
02全部仓库
| 仓库 | 分类 | ★ | 最新版本 | 许可证 |
|---|---|---|---|---|
| jax可组合变换的数值计算 | 训练与微调 | 36.5k | jax-v0.12.02026-10-19 | Apache-2.0 |
| scaling-book《How To Scale Your Model》：在 TPU 上扩展大模型的教材 | 学习资料 / Awesome 清单 | 1,491 | — | MIT |
| jax-tritonJAX 与 OpenAI Triton 的集成 | 算子 / 编译 / 硬件加速 | 472 | v0.4.12026-09-24 | Apache-2.0 |
| ml_dtypes机器学习常用 NumPy 数据类型扩展（bfloat16、fp8 等） | 算子 / 编译 / 硬件加速 | 363 | v0.6.02026-08-13 | Apache-2.0 |