dm-haiku
Haiku is a library for building neural networks in JAX.
Installation
To install this package, run one of the following:
Usage Tracking
0.0.16
0.0.15
0.0.14
0.0.13
0.0.12
5 / 8 versions selected
Downloads (Last 6 months): 0
Description
Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet, a neural network library for TensorFlow.
About
Summary
Haiku is a library for building neural networks in JAX.
Last Updated
Dec 17, 2025 at 19:21
License
Apache-2.0
Total Downloads
50.5K
Version Downloads
1.9K
Supported Platforms
noarch
GitHub Repository
https://dm-haiku.readthedocs.io/Documentation
https://github.com/deepmind/dm-haiku