Repository profile
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Repository updates
Get generated jax-ml/jax 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-18 04:14
53 captures since 2026-05-22
Stars from baseline +338
All tracked data
Frameworks, package managers, ecosystems, and dependency manifests found during catalog scans.
Scanned 2026-07-18 04:14
pyproject.toml
python ecosystem,
2 dependencies
setup.py
python ecosystem,
7 dependencies
docs/requirements.txt
python ecosystem,
26 dependencies
jaxlib/setup.py
python ecosystem,
3 dependencies
docs/ffi/CMakeLists.txt
c-cpp ecosystem,
1 dependency
examples/ffi/CMakeLists.txt
c-cpp ecosystem,
3 dependencies
examples/ffi/pyproject.toml
python ecosystem,
5 dependencies
jax_plugins/cuda/pyproject.toml
python ecosystem,
2 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/
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.
The pretty much "official" DSPy framework for Typescript
