Repository profile
Optax is a gradient processing and optimization library for JAX.
Repository updates
Get generated google-deepmind/optax development summaries by email, or follow the weekly and monthly RSS feeds.
Sign in to subscribe by email. RSS feeds are public.
Sign in to subscribeTracked growth, recent movement, and commit velocity from stored repository snapshots.
Latest capture 2026-07-07 03:15
6 captures since 2026-05-22
Stars from baseline +28
All tracked data
Frameworks, package managers, ecosystems, and dependency manifests found during catalog scans.
Scanned 2026-07-07 03:15
pyproject.toml
python ecosystem,
19 dependencies
docs/pyproject.toml
python ecosystem,
0 dependencies
examples/pyproject.toml
python ecosystem,
0 dependencies
Searchable topics, generated tags, and stack labels that explain where this repository fits.
Agent instructions and tool configuration paths found in the repository tree.
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
Go ahead and axolotl questions
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
