CommonLoopUtils

Jupyter Notebook ★ 367 updated 3mo ago

CLU lets you write beautiful training loops in JAX.

No plain-English explanation yet — one is being written right now. Check back in a minute.