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.stax

Browse Source 
The new location is jax.example_libraries.stax
This commit is contained in:
Jake VanderPlas 2022-08-01 14:58:41 -07:00
parent 41eff98de3
commit 91dbcbf525
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

@ -18,6 +18,8 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
for an alternative API. for an alternative API.
* {func}`jax.tree_util.tree_multimap` has been removed. It has been deprecated since * {func}`jax.tree_util.tree_multimap` has been removed. It has been deprecated since
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
{mod}`jax.example_libraries.stax`.
## 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

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

30
jax/experimental/stax.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.
"""Stax has moved to jax.example_libraries.stax
jax.experimental.stax is deprecated and will delegate to
jax.example_libraries.stax with a warning for backwards-compatibility
for a limited time.
"""
import warnings
from jax.example_libraries.stax import * # noqa: F401,F403
_HAS_DYNAMIC_ATTRIBUTES = True
warnings.warn('jax.experimental.stax is deprecated, '
'import jax.example_libraries.stax instead',
FutureWarning)