Rename jax.experimental.optimizers -> jax.example_libraries.optimizers

google-research / agent-based-epidemic-sim

Apache License 2.0

47 stars 17 forks source link

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

Rename jax.experimental.optimizers -> jax.example_libraries.optimizers

Why? The former name has been deprecated since JAX version 0.2.25, released in November 2021 (see https://github.com/google/jax/blob/main/CHANGELOG.md#jax-0225-nov-10-2021), and will soon be removed.