mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 20:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,158 Commits 502 Branches 263 Tags
Commit Graph

275 Commits

Author SHA1 Message Date
Skye Wanderman-Milne
1b87ee07c4 Update setup.py to automatically pick up libtpu patch releases 2025-02-27 17:22:29 -08:00
Peter Hawkins
85add667ed Change documentation to recommend libtpu from pypi instead of GCS. 2025-02-24 17:50:29 -05:00
Peter Hawkins
c8c4cfa04e Update version numbers after 0.5.1 release. 2025-02-24 16:18:25 -05:00
Yash Katariya
07440f4afa Prepare for JAX release 0.5.1 2025-02-24 10:59:04 -05:00
Peter Hawkins
b1a2c27aa0 Remove libtpu-nightly dependency from jax[tpu]. 
For several releases, libtpu-nightly has been a transitional empty package that does nothing. We remove the dependency in preparation for depending on libtpu from pypi instead of a GCS bucket in jax[tpu].
 2025-02-04 20:59:30 -05:00
Peter Hawkins
9fa2912254 Update version numbers after 0.5.0 release 2025-01-17 13:30:59 -05:00
Peter Hawkins
c25fb92c44 Release JAX 0.5.0 2025-01-17 10:28:03 -05:00
Robert Dyro
aa9cea0a55 Add quotes to pip commands in docs around option install for zsh 
PiperOrigin-RevId: 716358130
 2025-01-16 13:30:43 -08:00
Peter Hawkins
392a851769 Increase the minimum SciPy version to 1.11.1. 
(1.11.0 was yanked from PyPi because of licensing problems, so 1.11.1 is the oldest 1.11 release.)

PiperOrigin-RevId: 713073731
 2025-01-07 16:10:45 -08:00
Ruturaj4
fefebebea4 Update documentation and add setup.py pypi bindings 2024-12-19 10:21:28 -06:00
Peter Hawkins
ee45718457 Increase the minimum NumPy version to v1.25. 
Per SPEC 0, we drop NumPy v1.24 support on Dec 18, 2024.
 2024-12-18 08:18:57 -05:00
Peter Hawkins
ff52aedf67 Update version numbers after release. 2024-12-17 18:16:25 -05:00
Peter Hawkins
c4e3f5a356 Release v0.4.38 2024-12-17 14:35:08 -05:00
Peter Hawkins
820f51dc53 Merge branch 'release/0.4.37' into main. 2024-12-09 20:21:43 -05:00
Peter Hawkins
ffb07cdadb Update versions for v0.4.37 release. 2024-12-09 15:39:59 -05:00
Peter Hawkins
ba626fa650 Bump JAX version after release. 
PiperOrigin-RevId: 703472753
 2024-12-06 05:58:09 -08:00
Peter Hawkins
7e6620a577 JAX release 0.4.36. 2024-12-05 16:08:39 -05:00
Jake VanderPlas
a4266b5e31 Mention python 3.13 in docs & package metadata 2024-11-20 08:23:19 -08:00
Peter Hawkins
2aeda17829 Merge branch 'release/0.4.35' 2024-10-23 08:50:31 -04:00
Peter Hawkins
81991d87c8 JAX release 0.4.35 2024-10-22 15:00:23 -04:00
Peter Hawkins
e4f3f8f064 Use libtpu releases rather than libtpu-nightly for jax[tpu]. 
PiperOrigin-RevId: 688632409
 2024-10-22 11:47:07 -07:00
Jake VanderPlas
74fc7360f5 Require ml_dtypes >= 0.4.0 
This is the minimum version that supports NumPy 2.0.
 2024-10-22 09:04:28 -07:00
Peter Hawkins
b0b7a60e63 Merge branch 'release/0.4.34' 2024-10-04 10:56:18 -04:00
Peter Hawkins
affba367c5 JAX 0.4.34 release 2024-10-02 14:37:09 -04:00
Yu-Hang Tang
c88c3aecae add k8s cluster environment 2024-09-20 17:26:53 +00:00
Michael Hudgins
d4d1518c3d Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax 
PiperOrigin-RevId: 676843138
 2024-09-20 07:52:33 -07:00
Peter Hawkins
ae0e403c60 Merge release/0.4.33 into main and update version numbers. 2024-09-16 18:46:24 +00:00
Peter Hawkins
80e1c94de6 Prepare for v0.4.33 release. 
This release is branched off the v0.4.32 release, with two changes:
a) a fixed libtpu pin, and
b) a patch to revert an F64 tanh issue on CPU.
 2024-09-16 13:30:35 +00:00
Peter Hawkins
3e81ae530d Update version numbers after v0.4.32 release. 2024-09-11 16:18:56 -04:00
Peter Hawkins
1594d2f30f Prepare for v0.4.32 release. 2024-09-11 14:29:55 -04:00
Yash Katariya
0b04dd022a Reverts 5e4250e64bb415be94ddc8a80dba6083a6a4123a 
PiperOrigin-RevId: 673141373
 2024-09-10 16:01:00 -07:00
Yash Katariya
5e4250e64b Prepare for jax 0.4.32 release 
PiperOrigin-RevId: 673105544
 2024-09-10 14:26:17 -07:00
Yash Katariya
2106a25977 Finish jax and jaxlib v0.4.31 release 
PiperOrigin-RevId: 657388782
 2024-07-29 17:57:37 -07:00
Yash Katariya
7fd9302785 Start JAX and jaxlib 0.4.31 release 
PiperOrigin-RevId: 657295431
 2024-07-29 12:49:58 -07:00
Peter Hawkins
fd23b8733d Bump minimum SciPy version to 1.10. 
SciPy 1.9.0 was released July 29, 2022, which is 24 months ago

PiperOrigin-RevId: 657215038
 2024-07-29 08:50:18 -07:00
Peter Hawkins
945fde41e4 Update minimum Python version to 3.10. 2024-06-26 13:47:14 -04:00
Peter Hawkins
7f24837eef Update minimum NumPy version to v1.24. 2024-06-21 15:17:17 -07:00
Yash Katariya
045ea944c8 Finish jax and jaxlib 0.4.30 release 
PiperOrigin-RevId: 644402635
 2024-06-18 08:56:39 -07:00
Peter Hawkins
3f4f79c83e Prepare for 0.4.30 release. 2024-06-18 07:34:45 -04:00
Peter Hawkins
dcb7b3c3f1 Readd cuda12_pip extra to keep CI users happy. 2024-06-13 12:37:06 -04:00
Peter Hawkins
b13733c13f Update JAX dependencies, extras, and documentation for plugins. 
* Make jaxlib a direct dependency of jax.
* Remove mentions of monolithic CUDA installations from the JAX documentation.
* Drop the cuda12_pip extra and the cudnn version specific extras.
* Add a with_cuda extra to the jax-cuda12-plugin package, use it in jax's setup.py. This allows us to specify cuda extras in one place.
* Make a few small doc improvements.
 2024-06-13 11:36:23 -04:00
Yash Katariya
b1f7627c71 [Rollback] Bumped the minimum ml_dtypes version to 0.4.0 
Reverts e86c436e7f8e4e0546eff8bc2d3756a7c49dc83b

PiperOrigin-RevId: 642741832
 2024-06-12 14:40:40 -07:00
Peter Hawkins
6fa31e59c4 Update version numbers after v0.4.29 release. 2024-06-10 14:37:53 -04:00
Peter Hawkins
e071053d97 Prepare for 0.4.29 release. 2024-06-10 08:54:28 -04:00
Henning Becker
437263659c Fix pip dependency after cuDNN 9 upgrade 
PiperOrigin-RevId: 637959308
 2024-05-28 11:06:48 -07:00
Henning Becker
15a1985445 Update cuDNN to version 9.1.1 in JAX 
PiperOrigin-RevId: 636956696
 2024-05-24 10:10:21 -07:00
jax authors
e86c436e7f Merge pull request #21404 from superbobry:maint 
PiperOrigin-RevId: 636862158
 2024-05-24 03:53:58 -07:00
Parker Schuh
c1f5a32875 Drop refs to australis in build rules and exporter. 
PiperOrigin-RevId: 636720089
 2024-05-23 16:52:58 -07:00
Sergei Lebedev
0a694a1b42 Bumped the minimum ml_dtypes version to 0.4.0 2024-05-23 21:51:00 +01:00
Jake VanderPlas
9ac1d38226 Finish jax and jaxlib 0.4.28 release 
PiperOrigin-RevId: 632653310
 2024-05-10 18:06:52 -07:00