py-dm-haiku (spack): JAX-based neural network library 
Keywords: deep-learning|deep-neural-networks|jax|machine-learning|neural-networks