Optax

E438357 UNEXPLORED

Optax is a gradient processing and optimization library for JAX, providing a flexible collection of composable optimizers and transformations for training machine learning models.


Referenced by (1)
Subject (surface form when different) Predicate
JAX
compatibleWith

Please wait…