A Pytree Module system for Deep Learning in JAX
Primary LanguagePythonMIT LicenseMIT
No one’s watching this repository yet.