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

Merge pull request #12511 from hawkinsp:release

Browse Source 
PiperOrigin-RevId: 476889960
This commit is contained in:
jax authors 2022-09-26 07:24:44 -07:00
commit 2180710c2a
3 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
CHANGELOG.md
View File

@ -8,8 +8,8 @@ Remember to align the itemized text with the first line of an item within a list
PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
-->
## jax 0.3.18 (Unreleased)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.17...main).
## jax 0.3.18 (Sep 26, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.17...jax-v0.3.18).
* Changes
* Ahead-of-time lowering and compilation functionality (tracked in
{jax-issue}`#7733`) is stable and public. See [the
@ -22,6 +22,9 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
`jax.soft_pmap` is undocumented. If it were documented, a deprecation period
would have been provided.
## jaxlib 0.3.18 (Sep 26, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.15...jaxlib-v0.3.18).
## jax 0.3.17 (Aug 31, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.16...jax-v0.3.17).
* Bugs
@ -62,9 +65,6 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
implementation switched on by default, meaning the old implementation is
deprecated; see [JEP 11830](https://jax.readthedocs.io/en/latest/jep/11830-new-remat-checkpoint.html).
## jaxlib 0.3.16 (Unreleased)
* [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.15...main).
## jax 0.3.15 (July 22, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.14...jax-v0.3.15).
* Changes

6
WORKSPACE
View File

@ -7,10 +7,10 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
# and update the sha256 with the result.
http_archive(
name = "org_tensorflow",
sha256 = "9073ab3cbf3a89baee459f6e953cee240864393774f568fdba200a6ff5512c9f",
strip_prefix = "tensorflow-a4905aa04186bcaf89b06032baa450cc1ce103ad",
sha256 = "23e1cf573ee3857f4e598dd14ae2c1554d72fbf6bcec500c6315aa930865eddf",
strip_prefix = "tensorflow-8dcaf6b98a6a49c85eb470140ba8506e71a3b5af",
urls = [
"https://github.com/tensorflow/tensorflow/archive/a4905aa04186bcaf89b06032baa450cc1ce103ad.tar.gz",
"https://github.com/tensorflow/tensorflow/archive/8dcaf6b98a6a49c85eb470140ba8506e71a3b5af.tar.gz",
],
)

4
setup.py
View File

@ -19,14 +19,14 @@ import sys
from setuptools import setup, find_packages
_current_jaxlib_version = '0.3.15'
_current_jaxlib_version = '0.3.18'
# The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.3.15'
_available_cuda_versions = ['11']
_default_cuda_version = '11'
_available_cudnn_versions = ['82', '805']
_default_cudnn_version = '82'
_libtpu_version = '0.1.dev20220723'
_libtpu_version = '0.1.dev20220926'
_dct = {}
with open('jax/version.py') as f: