Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Quick read
Latest capture 2026-10-17 04:23
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.
143 observed captures since 2026-05-22. Observed captures are shown by default.
Stars from first capture +781
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/
JAX-based neural network library
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Optax is a gradient processing and optimization library for JAX.
Flax is a neural network library for JAX that is designed for flexibility.
Adaptive Experimentation Platform
