google-deepmind/dm-haiku
JAX-based neural network library
A Graph Neural Network Library in Jax
JAX-based neural network library
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Build Graph Nets in Tensorflow
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/
3 captures since 2026-05-22