dm-haiku

Python ★ 3.2k updated 18d ago

JAX-based neural network library

No plain-English explanation yet — one is being written right now. Check back in a minute.