GitHub repository: sshyran/haiku
Description: JAX-based neural network library