dm-haiku

JAX-based neural network library

3k stars

38 watching

232 forks

Language: Python

last commit: 5 days ago

Linked from 1 awesome list

deep-learningdeep-neural-networksjaxmachine-learningneural-networks

Screenshot of google-deepmind/dm-haiku website

https://dm-haiku.readthedocs.io

Backlinks from these awesome lists:

n2cholas/awesome-jax