Merge pull request #11712 from jakevdp:delete-optimizers

PiperOrigin-RevId: 466398079
google · Aug 9, 2022 · 88636d2 · 88636d2
2 parents d95b27c + 7940675
commit 88636d2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 31 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -26,6 +26,8 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
     JAX release 0.3.5, and {func}`jax.tree_util.tree_map` is a direct replacement.
   * Removed `jax.experimental.stax`; it has long been a deprecated alias of
     {mod}`jax.example_libraries.stax`.
+  * Removed `jax.experimental.optimizers`; it has long been a deprecated alias of
+    {mod}`jax.example_libraries.optimizers`.
 
 ## jaxlib 0.3.16 (Unreleased)
 * [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.15...main).

diff --git a/jax/BUILD b/jax/BUILD
@@ -159,7 +159,6 @@ pytype_library(
     name = "optimizers",
     srcs = [
         "example_libraries/optimizers.py",
-        "experimental/optimizers.py",
     ],
     visibility = ["//visibility:public"],
     deps = [":jax"],

diff --git a/jax/experimental/optimizers.py b/jax/experimental/optimizers.py