dm-haiku (pypi): Haiku is a library for building neural networks in JAX.
Keywords: deep-learning|deep-neural-networks|jax|machine-learning|neural-networks