Jump to content

View source for Jax (library)

← Jax (library)

You do not have permission to edit this page, for the following reason:

The action you have requested is limited to users in the group: Users.

You can view and copy the source of this page.

Notes about Jax

==Install==
===Conda===
<pre>
conda install jax jaxlib=*=*cuda* -c conda-forge
</pre>
===pip===
<pre>
pip install --upgrade jax jaxlib==0.3.10+cuda11.cudnn82 -f https://storage.googleapis.com/jax-releases/jax_releases.html
</pre>

Return to Jax (library).

Retrieved from "https://wiki.davidl.me/view/Jax_(library)"