nnx

Hyperparameter Scheduling in JAX/Flax with Optax: Use Optimizer Step to Drive Schedules (Flax 0.11)

Sep. 21, 17:00

1