JAX-based neural network library
google-deepmind/dm-haiku has 3,254 GitHub stars.
It currently ranks #20532 daily, #21288 weekly, #23865 monthly by momentum on Trending Repos; its best-ever rank is #12846.