flax

Kevin Murphy《Probabilistic Machine Learning》一书的 Python 代码

Flax Engine – 多平台 3D 游戏引擎

阿姆斯特丹大学（MSc AI）2023 秋季深度学习课程的 Jupyter Notebook 教学资料库。

让大型语言模型（LLM）变得简单，EasyLM 是在 JAX/Flax 中进行 LLM 预训练、微调、评估与服务的一站式解决方案。

模块化强化学习（RL）库（使用 PyTorch、JAX 与 NVIDIA Warp 实现），支持 Gymnasium/Gym、NVIDIA Isaac Lab、MuJoCo Playground 等环境。

Tevatron：跨规模、语言与模态的统一文档检索工具包。于 SIGIR 2023、SIGIR 2025 展示。

Orbax 为 JAX 用户提供通用的检查点与持久化工具