# google-deepmind/dm-haiku

JAX-based neural network library

Repository: https://github.com/google-deepmind/dm-haiku
Canonical: https://ross.abutalabs.com/products/dm-haiku
Homepage: https://dm-haiku.readthedocs.io
Language: Python
License: Apache-2.0
License Family: permissive
Topics: machine-learning, neural-networks, jax, deep-learning, deep-neural-networks
Last push: 2026-08-25T11:52:21+00:00

## Health v2 (maintenance only)
Score: 89/100 (v2, computed 2026-09-02T17:46:02.011165+00:00)
- activity 99, release rhythm 71, longevity 100
- inputs: {"age_days": 2388, "days_push": 8, "days_rel": 37, "gap_med": 168.5, "n_releases_24m": 5}
- flags: none
- formula: round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)

## Adoption (not part of the score)
Stars 3275, forks 298 (observed 2026-08-28T04:07:53.443271+00:00)

## What it is
Haiku is a simple neural network library for JAX from Google DeepMind, providing an object-oriented module abstraction (hk.Module) and a function transformation (hk.transform) that bridges OOP-style code with JAX's pure function transformations. It is the JAX counterpart of Sonnet and is now in maintenance mode, with DeepMind recommending Flax for new projects.

## Use cases
- build neural networks in jax with object-oriented modules
- convert oop-style model code into pure jax functions
- port sonnet models from tensorflow to jax
- train deep learning models with gpu/tpu support in jax
- manage model parameters in jax research code

## When to choose
- you maintain existing Haiku-based codebases
- you prefer Sonnet-style OOP module APIs in JAX
- you need a minimal, stable neural network library for JAX research

## When to avoid
- you are starting a new project - DeepMind recommends Flax
- you need new features or an active community
- you want extensive end-to-end examples and documentation

## Facets
- artifact type: library
- maturity: maintenance
- function: machine-learning, deep-learning
- domain: machine-learning, deep-learning
- platform: python
- tags: jax, neural-networks, sonnet, google-deepmind, object-oriented-api, gpu

## Member repositories
- google-deepmind/dm-haiku (main) score 89

## Provenance
- Observed fields: from GitHub, fetched 2026-08-28T04:07:53.443271+00:00.
- Health v2: computed from the inputs above; adoption is never an input.
- Inferred fields (summary, facets, guidance): AI-extracted, prompt v1, taxonomy v1, on 2026-08-29T18:43:22.899573+00:00, confidence not recorded.
  - readme: https://github.com/google-deepmind/dm-haiku (fetched 2026-08-28T04:07:53.443271+00:00, sha e445d3beb9d7)
- Data as of 2026-08-30T08:39:29.467469+00:00.