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