JAX-based neural network library
Appears on
Quick read
Latest capture 2026-09-09 03:07
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.
7 observed captures since 2026-05-25. Observed captures are shown by default.
Stars from first capture +44
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
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Flax is a neural network library for JAX that is designed for flexibility.
A Graph Neural Network Library in Jax
Optax is a gradient processing and optimization library for JAX.
Go ahead and axolotl questions
