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