google-deepmind / dm-haiku

JAX-based neural network library

GabrielNicolas10 / haiku-forge

HaikuTrainer Pro 2026: Best JAX Library for Fast Haiku Model Training

Gabedevz / haiku-bridge-trainer

Bax: Flexible JAX and Haiku Trainer Interface for Scalable Deep Learning 2026