Rare find

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

github.com/google-deepmind/optax

@google-deepmind

Problem

Audience

Updates

No founder updates yet. Check back soon.