Jax (library): Difference between revisions
Created page with "Notes about Jax ==Install== Through conda: <pre> conda install jax jax=*=*cuda* -c conda-forge </pre>" |
|||
| Line 4: | Line 4: | ||
Through conda: | Through conda: | ||
<pre> | <pre> | ||
conda install jax | conda install jax jaxlib=*=*cuda* -c conda-forge | ||
</pre> | </pre> | ||
Revision as of 21:06, 27 May 2022
Notes about Jax
Install
Through conda:
conda install jax jaxlib=*=*cuda* -c conda-forge