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
25,122 Commits 502 Branches 263 Tags
Commit Graph

208 Commits

Author SHA1 Message Date
Peter Hawkins
3a8f31aa83 Update the JAX version to 0.5.0. 
This is because of the breaking change to PRNG key semantics, and the version follows JAX's new effver versioning scheme (https://jax.readthedocs.io/en/latest/jep/25516-effver.html).
 2025-01-15 14:08:15 -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
Yunlong Liu
27bf80a506 Adds an env that can let users provide a custom version suffix for jax dev build. 
fix the local version

update to what Jake suggested
 2024-11-16 14:39:54 -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
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
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
46a7be6d7a Lower the minimum jaxlib version to 0.4.31, since 0.4.32 was yanked from pypi. 
PiperOrigin-RevId: 673992844
 2024-09-12 13:46:23 -07: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
30037547d7 Bump minimum jaxlib version to 0.4.31. The corresponding xla_extension_version is 279 and mlir_api_version is 57 
PiperOrigin-RevId: 657400413
 2024-07-29 18:44:31 -07:00
Yash Katariya
2106a25977 Finish jax and jaxlib v0.4.31 release 
PiperOrigin-RevId: 657388782
 2024-07-29 17:57:37 -07:00
Peter Hawkins
07d24e7dcc Bump minimum jaxlib version to v0.4.30. 
This corresponds to xla_extension_version 271 and mlir_api_version 57.
 2024-06-18 12:35:08 -04: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
6fa31e59c4 Update version numbers after v0.4.29 release. 2024-06-10 14:37:53 -04:00
Jake VanderPlas
9ac1d38226 Finish jax and jaxlib 0.4.28 release 
PiperOrigin-RevId: 632653310
 2024-05-10 18:06:52 -07:00
Yash Katariya
395d3cb79e Bump minimum jaxlib version to 0.4.27 
xla_extension_version is 261 and mlir_api_version is 56

PiperOrigin-RevId: 631579739
 2024-05-07 16:07:59 -07:00
Yash Katariya
5031a1ddc4 Finish jax and jaxlib 0.4.27 release 
PiperOrigin-RevId: 631486157
 2024-05-07 11:14:09 -07:00
Sergei Lebedev
754fab91f7 Bumped the minimum jaxlib to 0.4.23 
jaxlib 0.4.23 has xla_extension_version 223 and mlir_api_version 54.
 2024-04-13 08:18:33 +01:00
Olli Lupton
b27b77446f _version_from_git_tree: avoid git describe 2024-04-10 12:47:47 +00:00
Yash Katariya
24517ca3e0 Finish jax and jaxlib 0.4.26 release 
PiperOrigin-RevId: 621658207
 2024-04-03 15:40:24 -07:00
Yash Katariya
e0fd29082d Finish jax and jaxlib 0.4.25 release 
PiperOrigin-RevId: 610413312
 2024-02-26 08:19:05 -08:00
Peter Hawkins
aad02dba7e Increase minimum jaxlib version to 0.4.20. 
jaxlib 0.4.20 has xla_extension_version 210 and mlir_api_version 54.

PiperOrigin-RevId: 609094229
 2024-02-21 12:58:57 -08:00
Skye Wanderman-Milne
b93772fc24 Update version numbers post-0.4.24 release 2024-02-06 16:28:23 -08:00
Parker Schuh
23b9c2a22f Add the githash that the jaxlib was built at to __init__.py. This is to allow identifying the githash of nightlies. 
PiperOrigin-RevId: 595529249
 2024-01-03 16:12:23 -08:00
Yash Katariya
6e1ab7ca3f Finish release of jax and jaxlib 0.4.23 
PiperOrigin-RevId: 590833947
 2023-12-13 23:39:08 -08:00
Yash Katariya
25c16c0b78 Finish jax and jaxlib 0.4.22 release 
PiperOrigin-RevId: 590775311
 2023-12-13 18:26:47 -08:00
Yash Katariya
a9bfbd32e1 Finish jax and jaxlib 0.4.21 release 
PiperOrigin-RevId: 587866580
 2023-12-04 15:51:58 -08:00
Olli Lupton
e50c35d1e7 Fix repeatedly building JAX. 
Reproducer was essentially running `pip install .` twice in a row in the
same source directory. Closes google/jax#18252.
 2023-11-30 13:31:17 +01:00
Jake VanderPlas
cc9656d654 version: fix commit hash extraction 2023-11-22 09:35:03 -08:00
Peter Hawkins
30a0136813 Increase minimum jaxlib version to 0.4.19. 
0.4.19 has xla_extension version 207 and mlir_api_version 54.

PiperOrigin-RevId: 583412447
 2023-11-17 09:38:31 -08:00
Skye Wanderman-Milne
55e3072d2e Update versions and CHANGELOG after jax 0.4.20 release 2023-11-02 16:30:56 -07:00
Yash Katariya
613369fc22 Finish 0.4.19 jax and jaxlib release 
PiperOrigin-RevId: 574983871
 2023-10-19 13:27:52 -07:00
Skye Wanderman-Milne
a06beaa1a2 Update versions post jax 0.4.18 release 2023-10-06 17:20:34 -07:00
Skye Wanderman-Milne
82b58386b7 Update versions and CHANGELOG after jax 0.4.17 release 2023-10-03 17:54:35 -07:00
Jake VanderPlas
4a5bd9e046 Fix typos across the package 2023-09-22 14:54:31 -07:00
Yash Katariya
dcc465b4de Finish jax and jaxlib 0.4.16 release 
PiperOrigin-RevId: 566477931
 2023-09-18 19:09:19 -07:00
Jake VanderPlas
db6104a3dd Build: catch stderr when attempting git commands 2023-09-08 10:15:55 -07:00
jax authors
f64235acc8 Merge pull request #17453 from jakevdp:fix-version-string 
PiperOrigin-RevId: 563466394
 2023-09-07 10:06:32 -07:00
Jake VanderPlas
6f3f0d5e57 build: write appropriate version strings to build artifacts 2023-09-07 08:45:48 -07:00