google-deepmind/dm-haiku
JAX-based neural network library
A Graph Neural Network Library in Jax
Appears on
Quick read
Latest capture 2026-08-22 03:08
0 paths
Agent instructions and tool configuration found in this repository.
No config files detected.
8 observed captures since 2026-05-22. Observed captures are shown by default.
Stars from first capture +3
Observed captures only
All tracked data
Observed snapshots
Observed snapshots
Nearest indexed repositories by embedding similarity.
JAX-based neural network library
Build Graph Nets in Tensorflow
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Flax is a neural network library for JAX that is designed for flexibility.
JanusGraph: an open-source, distributed graph database
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/