>
7 articles tagged "optimization" — page 1 of 1
optax(github.com)
|tool|GitHub

**Optax** is a gradient processing and optimization library designed for use with JAX, a high-performance numerical computing framework. It provides a flexible and composable set of optimization algor...