deepmind/dm-haiku: JAX-based neural network library
Category: ml-dl
Language: Python
Topics: deep-learning, deep-neural-networks, jax, machine-learning, neural-networks
Homepage: https://dm-haiku.readthedocs.io