dm-haiku
by google-deepmind · GitHub
JAX-based neural network library
⭐ Stars
3.3k
🍴 Forks
300
🔥 Trending
+1 this week
📜 License
Apache-2.0
Commercial use OK📅 Created
2020
🔄 Last commit
12 days ago
🏷️ Category
deep-learning
💻 Language
You maintain this project?
Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.
📈 Star history
2026-07-072026-07-22
📄 About
JAX-based neural network library
❓ Frequently asked questions
What is dm-haiku?
JAX-based neural network library
Is dm-haiku open source?
dm-haiku is an open-source project. It is released under the Apache-2.0 license.
Is dm-haiku free?
Yes. dm-haiku is free and open source — you can use, modify and self-host it.
🏅 Maintainer of this project?
Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.
[](https://olud.ai/project/google-deepmind-dm-haiku.html)
🧬 Shares DNA with🧬 View the DNA map →
axlearn
An Extensible Deep Learning Library
★ 2.4k · deep-learning
sharesjax
keras
Deep Learning for humans
★ 64.2k · data-science
sharesjax
scenic
Scenic: A Jax Library for Computer Vision Research and Beyond
★ 3.8k · attention
sharesjax
Measured from GitHub topics shared by both projects, weighted by how rare each topic is.