optax

Python ★ 2.3k updated 3d ago

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.