dm-haiku has 750K monthly downloads and 3K GitHub stars. Version 0.0.16 supports Python 3.10+, first published in 2020.
It is actively maintained under a permissive license, last released in Dec 2025.
It is built on top of absl-py, jmp and numpy. It is used by 49 tracked packages, including alphafold-colabfold, bioemu and neuralgcm.
GitHub topics include machine-learning, neural-networks and jax.
Package Insights
- No releases in 197 days despite active commits
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Used By
- alphafold-colabfold
- bioemu
- neuralgcm
- sbijax
- surjectors
- alphapulldown
- mlip
- imt-ring
- ai-models-graphcast-gfs
- ai-models-graphcast
- ai-models-gencast
- dm-clrs
- gpax
- juxtapose
- sabr-kit
- alphagenome-ft
- deepqmc
- avici
- jax-relax
- disentangled-rnns
- tracr-pypi
- spyx
- jaxrie
- leonardo-toolset
- dmpe
- chemtrain
- alpharing
- immrax
- onescience
- causal-discovery
- lagrangebench
- kagglefold
- xpag
- neuroflex
- supergrad
- nndp
- haiku-mup
- safejax
- folding
- dopamax
- ludax-agents
- jaxex
- x-mlps
- mubelnet
- non-param-score-est
- bayes-jones
- nextgenjax
- hijax
- id-mlip