{"categories":[{"label":"Libraries","url":"https://skillfed.io/packages/category/software-development-libraries/7"},{"label":"Python Modules","url":"https://skillfed.io/packages/category/software-development-libraries-python-modules/14"},{"label":"Mathematics","url":"https://skillfed.io/packages/category/scientific-engineering-mathematics/2"}],"enrichment":{"capability":"Haiku is a neural network library for JAX that provides an object-oriented module abstraction and function transformation to build and train models while retaining access to JAX's pure function transformations.","skillfed_tags":["jax-ecosystem","maintenance-mode","neural-networks"],"use_cases":["Building and training image classification models with parameter management and automatic differentiation.","Implementing reinforcement learning agents where you need state management with functional transformations.","Migrating existing Sonnet/TensorFlow neural network code to JAX with minimal API changes.","Prototyping generative models that benefit from functional programming patterns.","Research projects where DeepMind has validated Haiku's reliability in large-scale experiments."],"what_it_does":"Haiku is a neural network library built on JAX that lets you write models using familiar object-oriented patterns\u2014defining modules with parameters and methods\u2014while automatically transforming them into pure functions compatible with JAX's transformations. It provides two core abstractions: `hk.Module` for encapsulating network state and computation, and `hk.transform` for converting module-based code into pure `init` and `apply` functions.\n\nThe library is designed to be minimal and composable, handling parameter initialization and state management without imposing custom optimizers, checkpointing formats, or replication APIs. It draws its API and programming model from Sonnet, making it familiar to users migrating from TensorFlow. As of July 2023, Google DeepMind recommends new projects adopt Flax instead; Haiku now operates in maintenance mode, receiving bug fixes and compatibility updates but no new features.","worth_installing":"Yes, if you are already committed to JAX and value a lightweight, Sonnet-like API for parameter management. No, if you are starting a new project\u2014Google DeepMind officially recommends Flax instead, which has more active development and broader adoption. Haiku remains best-effort supported indefinitely but will not gain new features; install it only when migrating existing code or when its specific design philosophy is a deliberate fit."},"id":"dm-haiku","links":{"html":"https://skillfed.io/packages/dm-haiku","md":"https://skillfed.io/packages/dm-haiku.md","pypi":"https://pypi.org/project/dm-haiku/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2026-07-27","license_spdx":null,"license_treatment":"permissive","name":"dm-haiku","python_support":"supports_current","summary":"Haiku is a library for building neural networks in JAX."},"popularity":{"monthly_downloads":262617,"position":8370,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.0.17"}