google-deepmind/dm-haiku
JAX-based neural network library
3234 stars
Python
1 awesome list
google-deepmind/jraph
A Graph Neural Network Library in Jax
- Stars
- 1,471
- Forks
- 105
- Commits
- 103
- Language
- Python
- Awesome lists
- 1
Similar repositories
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
35740 stars
Python
3 awesome lists
google-deepmind/graph_nets
Build Graph Nets in Tensorflow
5402 stars
Python
1 awesome list
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
7206 stars
Jupyter Notebook
2 awesome lists
JanusGraph/janusgraph
JanusGraph: an open-source, distributed graph database
5792 stars
Java
1 awesome list
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2888 stars
Python
1 awesome list
Tracked growth
3 captures since 2026-05-22
Latest capture 2026-05-29 03:00
Stars history
Total starsCommits history
Default branch commitsMetadata
- Created: 2020-11-23
- First commit: 2020-10-20
- Last pushed: 2024-03-18
- Website: https://jraph.readthedocs.io/en/latest/
- Archived: yes
- Stack detected: —
- License: Apache-2.0
AI development signals
No AI development config files detected.