mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
10,466 Commits 506 Branches 263 Tags
Commit Graph

75 Commits

Author SHA1 Message Date
Yash Katariya
2162868ed9 Update values after release 
PiperOrigin-RevId: 427910510
 2022-02-10 20:32:53 -08:00
Yash Katariya
1ad3551ec9 Release jax and jaxlib 0.3.0 as per the new release process. 
PiperOrigin-RevId: 427809845
 2022-02-10 11:59:13 -08:00
Peter Hawkins
6791446bb1 Update development jaxlib version to 0.1.77, update jaxlib version in setup.py to 0.1.76. 
Changelog entry for jaxlib 0.1.77 was already added in a previous PR.

PiperOrigin-RevId: 424872047
 2022-01-28 08:10:58 -08:00
Yash Katariya
1b5630eed6 Update jaxlib version number to 0.1.76 
PiperOrigin-RevId: 415050863
 2021-12-08 11:14:12 -08:00
Peter Hawkins
7902ddaca2 Update jaxlib versions. 2021-11-17 11:46:41 -05:00
Yash Katariya
4d8bce1b85 Add a default cuda installation path and more explicit installation paths for CUDA jaxlib. 
```
# Installs Cuda 11 with Cudnn 8.2
$ pip install jax[cuda] -f https://storage.googleapis.com/jax-releases/jax_releases.html

$ pip install jax[cuda=11,cudnn=82] -f https://storage.googleapis.com/jax-releases/jax_releases.html

$ pip install jax[cuda=11,cudnn=805] -f https://storage.googleapis.com/jax-releases/jax_releases.html
```

PiperOrigin-RevId: 404134291
 2021-10-18 19:56:22 -07:00
Skye Wanderman-Milne
0072c32546 Update CHANGELOG and verson numbers for jaxlib 0.1.72 release 2021-10-12 17:37:29 -07:00
yashkatariya
be824a792e Update files after new jaxlib release 0.1.71 2021-09-01 10:43:20 -07:00
Yash Katariya
bf967d88d8 Upgrade versions after jaxlib release 
PiperOrigin-RevId: 389753047
 2021-08-09 16:37:44 -07:00
Qiao Zhang
82e74959fe Update changelog for jaxlib-0.1.69. 2021-07-12 12:06:41 -07:00
Qiao Zhang
a22841b6bb Bump jaxlib ver to 0.1.68. 2021-06-23 12:37:56 -07:00
George Necula
6a48c60a72 Rename master to main in embedded links. 
Tried to avoid the change on external links to repos that
have not yet renamed master.
 2021-06-18 10:00:01 +03:00
Skye Wanderman-Milne
63dbb99a66 Update README, etc. for jaxlib 0.1.67 release 2021-05-17 17:48:46 -07:00
Qiao Zhang
528d5bbb11 Update README etc for jaxlib 0.1.66 release. 2021-05-11 16:49:32 -07:00
Skye Wanderman-Milne
f8f373466c Update README, etc. for jaxlib 0.1.65 release 2021-04-07 17:51:20 -07:00
Jake VanderPlas
f9a4162551 Specify minimum jaxlib version in a single location 2021-03-22 16:14:41 -07:00
Skye Wanderman-Milne
0cbe2c1c05 Update README, etc. for jaxlib 0.1.64 release 2021-03-18 16:11:40 -07:00
Skye Wanderman-Milne
757247b791 Update README, etc. for jaxlib 0.1.63 release 2021-03-17 10:14:52 -07:00
Skye Wanderman-Milne
f06bb9a7f4 Update jaxlib version etc. 2021-03-09 17:55:40 -08:00
Skye Wanderman-Milne
7a67b974ac jaxlib version bump etc. 2021-02-12 09:42:04 -08:00
Peter Hawkins
13f3819054 Update README.md for jaxlib 0.1.60. 
Bump jaxlib version to 0.1.61 and update changelog.

Change jaxlib numpy version limit to >=1.16 for next release. Releases older than 1.16 are deprecated per NEP 00029. Reenable NumPy 1.20.

Bump minimum jaxlib version to 0.1.60.
 2021-02-03 20:44:01 -05:00
Skye Wanderman-Milne
7c2454e969 Update jaxlib version, minimum jaxlib version, readme, and changelog. 
Bumping the min jaxlib version to support https://github.com/google/jax/pull/5213.
 2021-01-15 12:56:08 -08:00
Skye Wanderman-Milne
62b864a654 Update WORKSPACE and jaxlib version to 0.1.59 2021-01-14 13:38:17 -08:00
Skye Wanderman-Milne
36799f5007 README etc. updates for new jaxlib release 2020-11-12 13:48:44 -08:00
Skye Wanderman-Milne
a082169642 Update README and jaxlib.__version__ for new jaxlib release 2020-10-20 11:46:48 -07:00
Alexey Radul
537427aae2 Copybara import of the project: 
--
138105a9ea44e7a8c3ce575a4e51b7ed51518d41 by Skye Wanderman-Milne <skyewm@google.com>:

Update README, CHANGELOG, and jaxlib.__version__ for new jaxlib release

PiperOrigin-RevId: 338063494
 2020-10-20 08:25:23 -07:00
Skye Wanderman-Milne
138105a9ea Update README, CHANGELOG, and jaxlib.__version__ for new jaxlib release 2020-10-14 13:32:56 -07:00
Jean-Baptiste Lespiau
4ae3352efa Increase the version of jaxlib, to get the tests for the C++ jit running internally. 
The policy is the following:
- we do not create holes in the jaxlib release.
- it should thus be incremented only once between releases, but may not be incremented just before the release, e.g. to trigger internally some tests (that won't run on github).

PiperOrigin-RevId: 335938139
 2020-10-07 13:17:05 -07:00
Skye Wanderman-Milne
fc39332c4f
Clarify when jaxlib version should be bumped. (#4250) 2020-09-10 15:53:04 -07:00
Qiao Zhang
4600dd7957
Update jaxlib version for dlpack fix. (#4231) 2020-09-08 17:20:48 -07:00
Jean-Baptiste Lespiau
bdd65453b4
Add more features to the C++ jax.jit. (#4169) 
This mainly follows https://github.com/google/jax/pull/4089 by adding:

- support for disable_jit from C++
- support for jax._cpp_jit on methods.
- supporting applying @jax.jit on top-level functions, by delaying the retrieval of the device and backend.
- concurrency support.

I am not aware of any feature missing (but I suspect there are still some differences due to the differences between xla_computation and _xla_callable.)

See:

- https://i.ibb.co/ZMvZ4nK/benchmark.png for the benchmarking comparison (see
 cr/328899906 + benchmarks for how numbers were generated)
- The results of the Jax tests when enabling this:
http://sponge2/4a67d132-209f-45c5-ab7b-83716d329ec2 (110 fails, 92 passes, but many common cause of failure).
 2020-09-01 10:34:47 +03:00
Jean-Baptiste Lespiau
2ab6b42a45
Use pytree defined in tensorflow. (#4087) 
It also adds some tests on the scalar C++ conversion.
 2020-08-18 08:58:43 +03:00
George Necula
c7aff1da06
Revert "Use pytree from xla_client. (#4063)" (#4081) 
This reverts commit d8de6b61411179dcd2f63d7639bbcd69b30ac15f.

Tryting to revert because it seems that this produces test
failures in Google.
 2020-08-17 12:53:18 +03:00
Jean-Baptiste Lespiau
d8de6b6141
Use pytree from xla_client. (#4063) 2020-08-14 11:44:03 -04:00
Jake Vanderplas
17c21a79c3
update jaxlib version and changelog for pypi (#3824) 2020-07-22 15:21:30 -07:00
George Necula
a5ed161550
Release jaxlib 0.1.51 (#3636) 2020-07-02 10:33:00 +03:00
Peter Hawkins
2a6fc316c3
Update XLA in preparation for a new jaxlib release (0.1.50). (#3560) 2020-06-25 15:12:47 -04:00
Skye Wanderman-Milne
a088c023ff
Bump jaxlib version to 0.1.49 and update WORKSPACE (#3495) 2020-06-19 13:27:12 -07:00
Peter Hawkins
a06b122e4a
Add support for 64-bit FFTs. (#3290) 2020-06-02 09:41:44 -04:00
George Necula
e1cb032456
Prepare version 0.1.47 for jaxlib (#3008) 2020-05-08 10:01:28 +03:00
Peter Hawkins
9802d7321c
Update XLA. (#2927) 2020-05-01 21:08:56 -04:00
Skye Wanderman-Milne
4703b0fa3c
Bump jaxlib version to 0.1.45 and update WORKSPACE and CHANGELOG. (#2785) 2020-04-21 12:01:02 -07:00
Peter Hawkins
1bb67637ca
Add batch_group_count to conv_general_dilated. (#2635) 
* Add batch_group_count to conv_general_dilated.

* Use batch_group_count for RHS grouped convolution transpose rule.

* Implement lhs/rhs transpose and batching rules for batch_group_count convolution.
 2020-04-09 16:21:30 -04:00
Peter Hawkins
0c7e134cf4
Update XLA and increment jaxlib version to 0.1.43. (#2546) 2020-03-30 16:56:54 -04:00
Peter Hawkins
cecfb37e6c
Increment jaxlib version to 0.1.42. (#2457) 
Update XLA.
 2020-03-19 09:57:11 -04:00
Peter Hawkins
388e78f8e5
Increment jaxlib version to 0.1.41. (#2428) 
Update XLA.
 2020-03-16 11:24:02 -04:00
Peter Hawkins
9d61d4b616
Bump jaxlib version to 0.1.40. (#2352) 
Update XLA version.
 2020-03-04 14:49:11 -05:00
Skye Wanderman-Milne
c6b65c0a11 Bump jaxlib version to 0.1.39 and update WORKSPACE. 2020-02-11 18:58:28 -08:00
Skye Wanderman-Milne
f04348ed53 Bump jaxlib version to 0.1.38 and update WORKSPACE. 2020-01-21 16:59:27 -08:00
Skye Wanderman-Milne
12a62c1f33 Bump jaxlib version to 0.1.37 and update WORKSPACE. 2019-12-03 12:29:34 -08:00