mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
Copybara import of the project:
-- 391dea76bc8fe264cf26ec93d42147f87847894d by Peter Hawkins <phawkins@google.com>: Update version numbers after jax/jaxlib 0.3.7 release. COPYBARA_INTEGRATE_REVIEW=https://github.com/google/jax/pull/10324 from hawkinsp:jaxlib 391dea76bc8fe264cf26ec93d42147f87847894d PiperOrigin-RevId: 442311051
This commit is contained in:
parent
f2af76d0ad
commit
38ea5a6bc0
@ -13,6 +13,10 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
|
||||
* [GitHub
|
||||
commits](https://github.com/google/jax/compare/jax-v0.3.7...main).
|
||||
|
||||
## jaxlib 0.3.8 (Unreleased)
|
||||
* [GitHub
|
||||
commits](https://github.com/google/jax/compare/jaxlib-v0.3.7...main).
|
||||
|
||||
## jax 0.3.7 (April 15, 2022)
|
||||
* [GitHub
|
||||
commits](https://github.com/google/jax/compare/jax-v0.3.6...jax-v0.3.7).
|
||||
|
@ -409,7 +409,7 @@ pip install --upgrade "jax[cpu]"
|
||||
```
|
||||
|
||||
On Linux, it is often necessary to first update `pip` to a version that supports
|
||||
`manylinux2010` wheels.
|
||||
`manylinux2014` wheels.
|
||||
|
||||
### pip installation: GPU (CUDA)
|
||||
|
||||
|
@ -15,7 +15,7 @@
|
||||
def _version_as_tuple(version_str):
|
||||
return tuple(int(i) for i in version_str.split(".") if i.isdigit())
|
||||
|
||||
__version__ = "0.3.7"
|
||||
__version__ = "0.3.8"
|
||||
__version_info__ = _version_as_tuple(__version__)
|
||||
|
||||
_minimum_jaxlib_version = "0.3.2"
|
||||
|
@ -17,4 +17,4 @@
|
||||
# reflect the most recent available binaries.
|
||||
# __version__ should be increased after releasing the current version
|
||||
# (i.e. on main, this is always the next version to be released).
|
||||
__version__ = "0.3.7"
|
||||
__version__ = "0.3.8"
|
||||
|
2
setup.py
2
setup.py
@ -16,7 +16,7 @@ from setuptools import setup, find_packages
|
||||
|
||||
_current_jaxlib_version = '0.3.7'
|
||||
# The following should be updated with each new jaxlib release.
|
||||
_latest_jaxlib_version_on_pypi = '0.3.5'
|
||||
_latest_jaxlib_version_on_pypi = '0.3.7'
|
||||
_available_cuda_versions = ['11']
|
||||
_default_cuda_version = '11'
|
||||
_available_cudnn_versions = ['82', '805']
|
||||
|
Loading…
x
Reference in New Issue
Block a user