jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
JAX-based neural network library
Appears on
Quick read
Latest capture 2026-08-03 03:03
0 paths
Agent instructions and tool configuration found in this repository.
No config files detected.
6 observed captures since 2026-05-25. Observed captures are shown by default.
Stars from first capture +33
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