dm-haiku (conda): Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet, a neural network library for TensorFlow.
Keywords: deep-learning|deep-neural-networks|jax|machine-learning|neural-networks