Elegy is a framework-agnostic Trainer interface for the Jax ecosystem.
Primary LanguagePythonApache License 2.0Apache-2.0