/treex

A Pytree Module system for Deep Learning in JAX

Primary LanguagePythonMIT LicenseMIT

Watchers