optax

Python ★ 1 updated 11mo ago ⑂ fork

Optax is a gradient processing and optimization library for JAX.

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