A Pytree Module system for Deep Learning in JAX
Primary LanguagePythonMIT LicenseMIT
No issues in this repository yet.