Google's end-to-end machine learning platform with a complete ecosystem.
Flax
Flax is a neural network library for JAX that is designed for flexibility.
- Category
- ⚙️ Deep Learning Libraries
- Repository
- google/flax
- Stars
- ★ 7,349
- Forks
- 849
- Language
- Jupyter Notebook
- License
- Apache-2.0
- Last updated
- 2026-10-12
- Visits
- 0
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 训练与微调仓库。