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
26,292 Commits 506 Branches 263 Tags
Commit Graph

7 Commits

Author SHA1 Message Date
rajasekharporeddy
13774d1382 Fix Typos 2024-09-25 21:26:05 +05:30
Jake VanderPlas
09fd345de9 pre-commit: update hooks & pin using hashes 2024-08-27 15:23:13 -07:00
bion howard
1ace88bfba
Update quickstart.md 
fix minor grammar typo
 2024-07-09 12:48:51 -04:00
jax authors
fc1e1d4a65 Add freshness metablock to JAX OSS docs. 
PiperOrigin-RevId: 645508135
 2024-06-21 14:50:49 -07: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
Sam Ritchie
822bc3647d
Update quickstart to suggested jax[cuda12] installation 
This PR updates the quickstart docs to match the README and Installation Guide suggestion of using

```
pip install jax[cuda12]
```

for GPU.
 2024-05-28 13:31:14 -04:00
Jake VanderPlas
10ed827fe9 DOC: replace old tutorials with new content 2024-05-03 12:20:06 -07:00