
dm-haiku
by google-deepmind (organization)
JAX-based neural network library (description from the project’s GitHub page)
Repository facts
- Stars
- 3,295
- Forks
- 303
- Open issues and pull requests
- 114
- Primary language
- Python
- License
- Apache License 2.0 (Apache-2.0)
- Owner
- google-deepmind · Organization
- Created
- Feb 18, 2020
- Last push
- Oct 5, 2026
Topics
Related projects
Projects that share this one’s GitHub topics.
Rank 3,499: patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
2,114 starsPythonApache-2.0Last push Sep 6, 2026
Rank 5,355: pdebench/PDEBench
PDEBench: An Extensive Benchmark for Scientific Machine Learning
1,215 starsPythonOther licenseLast push Mar 30, 2026
Rank 7,719: netket/netket
Machine learning algorithms for many-body quantum systems
703 starsPythonApache-2.0Last push Oct 18, 2026
Rank 4,665: DLTK/DLTK
Deep Learning Toolkit for Medical Image Analysis
1,460 starsPythonApache-2.0Last push Mar 24, 2023