Optax is a gradient processing and optimization library for JAX.
Repository Index
64,265 visible repositories in the index
1–1 of 1 · Most stars
64,265 visible repositories in the index
1–1 of 1 · Most stars
Optax is a gradient processing and optimization library for JAX.