mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-21 22:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
14,698 Commits 508 Branches 263 Tags
Commit Graph

233 Commits

Author SHA1 Message Date
Owen Lockwood
2230225956 Update README.md 
Update README.md
 2023-01-06 15:01:43 -07:00
Peter Hawkins
7bd6cc13e0 Note CUDA 11.8 is required for Hopper and Ada Lovelace series GPUs. 
Fixes https://github.com/google/jax/issues/13570
 2022-12-12 11:03:48 -05:00
Yash Katariya
0118f8d568 Prepare for jax and jaxlib 0.4.0 release 
PiperOrigin-RevId: 493733609
 2022-12-07 16:02:24 -08:00
Jake VanderPlas
79c8d6679f README: fix badge URL 2022-12-07 08:48:40 -08:00
Jimmy
e3cd9671b9 Add x86_64 dependency note to pip installation 
Currently non x86_64 linux architectures are not supported, see #7097 for request to change this. This can lead to installation confusion, as jax will install, but jaxlib will not. For example see #12307. This adds a note to the install sections for the relevant pip wheels.
 2022-11-04 12:11:11 -07:00
Peter Hawkins
8f1a346198 Recommend using the cuda-nvcc package from the "nvidia" conda channel to fetch ptxas. 2022-08-16 13:34:07 -04:00
Peter Hawkins
026e760767 Point to the conda-forge jaxlib wheels in the JAX readme. 2022-08-15 20:34:09 +00:00
Peter Hawkins
a2c21958a5 Document multiprocess GPU support. 
Fixes #2731
 2022-08-09 11:31:05 -04:00
Matthew Johnson
3ab8637a73 Update README discussion of Windows support: 
* right next to the pip installation instructions, mention they don't work for Windows;
* add a link to #5795 for an unofficial discussion of Windows native support
 2022-07-19 17:12:40 -07:00
Yash Katariya
e780a402e8 Split cuda and nocuda index files into their own 
PiperOrigin-RevId: 455463113
 2022-06-16 14:07:55 -07:00
Roy Frostig
531ab3fb18 add CITATION.bib and bump version to latest release 
PiperOrigin-RevId: 453519227
 2022-06-07 14:00:21 -07:00
Filippo Vicentini
a076d349d9
Make CUDA install instructions copy-pasteablee 2022-05-11 17:57:26 +02:00
Dean Biskup
554b02e978
Add quotes to install commands in README 2022-04-19 12:23:52 -05:00
Peter Hawkins
38ea5a6bc0 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
 2022-04-16 22:37:09 -07:00
Yazeed Alaudah
4a8de96e88
Update README.md to fix link to prng.md 2022-02-20 09:46:49 -06:00
Peter Hawkins
d704c151fa Clarify the NVidia driver version requirements. 2022-02-17 14:37:29 -05:00
Peter Hawkins
384042d0e7 Clarify the required NVidia driver versions in the JAX readme instructions. 2022-01-28 11:35:34 -05:00
Jake VanderPlas
0a44aa69df README: clarify installation instructions for Colab TPU 2022-01-26 11:10:58 -08:00
Matthew Johnson
57f4fc9bbb update readme to focus on most active libraries 2021-11-15 13:47:13 -08:00
Brian Hulette
6cfbb890e6 Fix broken link to experimental/README.md 2021-10-20 13:01:26 -07:00
Roy Frostig
623c201054 [JAX] move example libraries from jax.experimental into jax.example_libraries 
The `jax.experimental.stax` and `jax.experimental.optimizers` modules are standalone examples libraries. By contrast, the remaining modules in `jax.experimental` are experimental features of the JAX core system. This change moves the two example libraries, and the README that describes them, to `jax.example_libraries` to reflect this distinction.

PiperOrigin-RevId: 404405186
 2021-10-19 17:30:45 -07:00
Yash Katariya
ee752b32f7 Use cuda11_cudnn82 instead of cuda=11,cudnn=82 because the latter one is a syntax error 
PiperOrigin-RevId: 404240654
 2021-10-19 06:24:53 -07: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
Yash Katariya
66a4a9ff3f Remove 10.2 cuda support 
PiperOrigin-RevId: 402707900
 2021-10-12 18:44:07 -07:00
Jake VanderPlas
d0e21072ec DOC: add brief comments to example in README 2021-09-14 11:47:19 -07:00
josephrocca
507cc99c84 Add a note about Google Colab setup 
User must run `jax.tools.colab_tpu.setup_tpu()`
 2021-09-13 21:34:02 -07:00
Yash Katariya
b603b34d5f Update the readme to reflect that Cuda 10.1 and cuda 11.0 support is dropped. 
PiperOrigin-RevId: 394772160
 2021-09-03 15:56:47 -07:00
Jake VanderPlas
e46200e123 README: mention that CUDA installation is for linux only 2021-08-28 08:15:21 -07:00
yashkatariya
1f13565b70 Remove opensource CS link since JAX is not indexed 2021-08-12 11:12:35 -07:00
elliotwaite
7392a57b75 DOC: many small fixes 2021-08-04 16:55:13 -07:00
Shawn Presser
f800029569
Fix broken links in README.md 2021-07-25 09:40:24 -05:00
Jamie Townsend
6ca775b10a
Add TPU precision details to README gotchas 2021-07-16 12:24:19 +02:00
Jake VanderPlas
5e2681939a DOC: update installation instructions for compatibility with zsh 2021-06-28 12:57:24 -07:00
Jake VanderPlas
3c727ab604 DOC: update install instructions for GPU & TPU 2021-06-24 11:22:12 -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
Peter Hawkins
f88e45295f Update installation instructions for CUDA wheels. 2021-05-12 09:24:31 -04: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
Lena Martens
25a2e1a66a
Fix the changelog link in the README. 2021-03-25 19:15:45 +00: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
Tom Hennigan
4f74b3391c
Update README.md 
Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com>
 2021-03-12 16:38:29 +00:00
Tom Hennigan
0fd83a30be Link to DeepMind JAX blog post. 2021-03-12 12:01:20 +00:00
Skye Wanderman-Milne
f06bb9a7f4 Update jaxlib version etc. 2021-03-09 17:55:40 -08:00
Jake VanderPlas
749ad95514 DOC: add transformations doc to HTML & reorganize contents 2021-03-08 16:25:04 -08:00
Jake VanderPlas
24e920e3f5 change colab.sandbox->colab.research 2021-02-25 10:29:43 -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