mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a link to JAX Toolbox containers

Browse Source
This commit is contained in:
Frederic Bastien 2023-08-10 08:29:58 -07:00
parent cf026ce745
commit f212634454
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@ -391,6 +391,8 @@ installed as the `jaxlib` package. Use the following instructions to install a
binary package with `pip` or `conda`, or to [build JAX from binary package with `pip` or `conda`, or to [build JAX from
source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source). source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).
There is also [containers](#containers).
We support installing or building `jaxlib` on Linux (Ubuntu 20.04 or later) and We support installing or building `jaxlib` on Linux (Ubuntu 20.04 or later) and
macOS (10.12 or later) platforms. There is also *experimental* native Windows macOS (10.12 or later) platforms. There is also *experimental* native Windows
support. support.
@ -583,6 +585,12 @@ See the `conda-forge`
[jax](https://github.com/conda-forge/jax-feedstock#installing-jax) repositories [jax](https://github.com/conda-forge/jax-feedstock#installing-jax) repositories
for more details. for more details.
### Containers
[JAX Toolbox](https://github.com/NVIDIA/JAX-Toolbox) has bleeding edge
GPU containers: nightly containers and models/frameworks
specific containers.
### Building JAX from source ### Building JAX from source
See [Building JAX from See [Building JAX from
source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source). source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).