Flax is a neural network library for JAX that is designed for flexibility.
Appears on
Quick read
Latest capture 2026-09-10 03:05
Archive status reflects the last catalog sync, not a maintenance assessment. Check recent releases, issue responses, and fit for your project with the repository shortlisting checklist.
0 paths
Agent instructions and tool configuration found in this repository.
No config files detected.
9 observed captures since 2026-05-22. Observed captures are shown by default.
Stars from first capture +108
Observed captures only
All tracked data
Observed snapshots
Observed snapshots
Nearest indexed repositories by embedding similarity.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
JAX-based neural network library
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Go ahead and axolotl questions
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.
