mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove deprecated jax.experimental.optimizers

Browse Source 
The new location is jax.example_libraries.optimizers
This commit is contained in:
Jake VanderPlas 2022-08-02 14:41:50 -07:00
parent ce80a54805
commit 79406757d0
3 changed files with 2 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CHANGELOG.md
View File

@ -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. 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 * Removed `jax.experimental.stax`; it has long been a deprecated alias of
{mod}`jax.example_libraries.stax`. {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) ## jaxlib 0.3.16 (Unreleased)
* [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.15...main). * [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.15...main).

1
jax/BUILD
View File

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

30
jax/experimental/optimizers.py
View File

@ -1,30 +0,0 @@
# Copyright 2021 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Optimizers have moved to jax.example_libraries.optimizers
jax.experimental.optimizers is deprecated and will delegate to
jax.example_libraries.optimizers with a warning for backwards-compatibility
for a limited time.
"""
import warnings
from jax.example_libraries.optimizers import * # noqa: F401,F403
_HAS_DYNAMIC_ATTRIBUTES = True
warnings.warn('jax.experimental.optimizers is deprecated, '
'import jax.example_libraries.optimizers instead',
FutureWarning)