Repository profile
JAX-based neural network library
Repository updates
Get generated google-deepmind/dm-haiku 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-15 03:12
5 captures since 2026-05-25
Stars from baseline +22
All tracked data
Frameworks, package managers, ecosystems, and dependency manifests found during catalog scans.
Scanned 2026-07-15 03:12
requirements-flax.txt
python ecosystem,
1 dependency
requirements-jax.txt
python ecosystem,
2 dependencies
requirements-test.txt
python ecosystem,
9 dependencies
requirements.txt
python ecosystem,
4 dependencies
setup.py
python ecosystem,
0 dependencies
docs/requirements.txt
python ecosystem,
7 dependencies
examples/requirements.txt
python ecosystem,
5 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.
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
The pretty much "official" DSPy framework for Typescript
Optax is a gradient processing and optimization library for JAX.
