Optax¶
Optax is a gradient processing and optimization library for JAX.
homepage: https://github.com/deepmind/optax
| version | versionsuffix | toolchain |
|---|---|---|
0.1.7 |
-CUDA-11.7.0 |
foss/2022a |
0.1.7 |
foss/2022a |
|
0.2.2 |
-CUDA-12.1.1 |
gfbf/2023a |
0.2.2 |
gfbf/2023a |
(quick links: (all) - 0 - a - b - c - d - e - f - g - h - i - j - k - l - m - n - o - p - q - r - s - t - u - v - w - x - y - z)