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

Merge pull request #16531 from skye:version

Browse Source 
PiperOrigin-RevId: 542722305
This commit is contained in:
jax authors 2023-06-22 18:16:00 -07:00
commit 60632980a8
3 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
CHANGELOG.md
View File

@ -6,7 +6,11 @@ Best viewed [here](https://jax.readthedocs.io/en/latest/changelog.html).
Remember to align the itemized text with the first line of an item within a list. Remember to align the itemized text with the first line of an item within a list.
--> -->
## jax 0.4.13 ## jax 0.4.14
## jaxlib 0.4.14
## jax 0.4.13 (June 22, 2023)
* Changes * Changes
* `jax.jit` now allows `None` to be passed to `in_shardings` and * `jax.jit` now allows `None` to be passed to `in_shardings` and
@ -38,7 +42,7 @@ Remember to align the itemized text with the first line of an item within a list
{func}`jax.experimental.jax2tf.convert` is deprecated in favor of the {func}`jax.experimental.jax2tf.convert` is deprecated in favor of the
new `native_serializaation_disabled_checks` ({jax-issue}`#16347`). new `native_serializaation_disabled_checks` ({jax-issue}`#16347`).
## jaxlib 0.4.13 ## jaxlib 0.4.13 (June 22, 2023)
* Changes * Changes
* Added Windows CPU-only wheels to the `jaxlib` Pypi release. * Added Windows CPU-only wheels to the `jaxlib` Pypi release.

2
jax/version.py
View File

@ -15,7 +15,7 @@
# This file is included as part of both jax and jaxlib. It is also # This file is included as part of both jax and jaxlib. It is also
# eval()-ed by setup.py, so it should not have any dependencies. # eval()-ed by setup.py, so it should not have any dependencies.
__version__ = "0.4.13" __version__ = "0.4.14"
_minimum_jaxlib_version = "0.4.11" _minimum_jaxlib_version = "0.4.11"
def _version_as_tuple(version_str): def _version_as_tuple(version_str):

2
setup.py
View File

@ -21,7 +21,7 @@ from setuptools import setup, find_packages
_current_jaxlib_version = '0.4.13' _current_jaxlib_version = '0.4.13'
# The following should be updated with each new jaxlib release. # The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.4.12' _latest_jaxlib_version_on_pypi = '0.4.13'
_available_cuda11_cudnn_versions = ['86'] _available_cuda11_cudnn_versions = ['86']
_default_cuda11_cudnn_version = '86' _default_cuda11_cudnn_version = '86'
_default_cuda12_cudnn_version = '89' _default_cuda12_cudnn_version = '89'