Google's end-to-end machine learning platform with a complete ecosystem.
JAX
可组合变换（JIT、vmap、grad）的高性能数值计算库。
- Category
- ⚙️ Deep Learning Libraries
- Repository
- jax-ml/jax
- Stars
- ★ 36,455
- Forks
- 3,861
- Language
- Python
- License
- Apache-2.0
- Last updated
- 2026-10-17
- Visits
- 0
Numerical computing
Similar projects
A unified interface to tens of thousands of pre-trained models covering text, images, audio and multimodal tasks.
A deep-learning framework with dynamic computation graphs, the mainstream choice in both research and industry.
The classic Python machine-learning library: simple and efficient.
多后端（JAX / TF / PyTorch）的深度学习 API。
最简单快速的中型 GPT 训练与微调仓库。