rwightman / optax

Optax is a gradient processing and optimization library for JAX.

Geek Repo

Github PK Tool