Optax is a gradient processing and optimization library for JAX.
Appears on
Quick read
Latest capture 2026-10-06 03:10
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 +74
Observed captures only
All tracked data
Observed snapshots
Observed snapshots
Nearest indexed repositories by embedding similarity.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Adaptive Experimentation Platform
JAX-based neural network library
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Go ahead and axolotl questions
