google-deepmind/dm-haiku
A neural network library for JAX developed by Google DeepMind, designed for building and training deep learning models.
Not currently ranked — collecting fresh signals.
star history
Haiku is a Python library that provides simple, composable primitives for building neural networks on top of JAX. It offers module abstractions, random number generation utilities, and common neural network components similar to Sonnet (for PyTorch) or Flax. The library is primarily used for defining, training, and evaluating deep learning models in research settings.
Frequently asked
- What is google-deepmind/dm-haiku?
- A neural network library for JAX developed by Google DeepMind, designed for building and training deep learning models.
- Is dm-haiku open source?
- Yes — google-deepmind/dm-haiku is open source, released under the Apache-2.0 license.
- What language is dm-haiku written in?
- google-deepmind/dm-haiku is primarily written in Python.
- How popular is dm-haiku?
- google-deepmind/dm-haiku has 3.3k stars on GitHub.
- Where can I find dm-haiku?
- google-deepmind/dm-haiku is on GitHub at https://github.com/google-deepmind/dm-haiku.