/CommonLoopUtils

CLU lets you write beautiful training loops in JAX.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Watchers

No one’s watching this repository yet.