/elegy

A High Level API for Deep Learning in JAX

Primary LanguagePythonMIT LicenseMIT

Watchers