GitHub repository: kp-forks/dm-haiku
Primary language: Python
Description: JAX-based neural network library