Home / JAX / Neural Network Libraries > Haiku Neural Network Libraries > Haiku Focused on simplicity, created by the authors of Sonnet at DeepMind. Package 3.2k stars GitHub Back to JAX