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