optax. Optax is a gradient processing and optimization library for JAX.

github.com/rwightman/optax
by @rwightman

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.