Revision 4d8bce1b8515c68a8432d379d964ea3e6bea463c authored by Yash Katariya on 19 October 2021, 02:55:51 UTC, committed by jax authors on 19 October 2021, 02:56:22 UTC
``` # 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
1 parent 6bd0c72
File | Mode | Size |
---|---|---|
_static | ||
_templates | ||
jax-101 | ||
notebooks | ||
sphinxext | ||
README.md | -rw-r--r-- | 133 bytes |
async_dispatch.rst | -rw-r--r-- | 5.5 KB |
autodidax.ipynb | -rw-r--r-- | 139.8 KB |
autodidax.md | -rw-r--r-- | 102.6 KB |
autodidax.py | -rw-r--r-- | 102.5 KB |
changelog.md | -rw-r--r-- | 33 bytes |
concurrency.rst | -rw-r--r-- | 609 bytes |
conf.py | -rw-r--r-- | 8.9 KB |
contributing.md | -rw-r--r-- | 6.3 KB |
custom_vjp_update.md | -rw-r--r-- | 4.9 KB |
deprecation.md | -rw-r--r-- | 528 bytes |
developer.md | -rw-r--r-- | 11.0 KB |
device_memory_profiling.md | -rw-r--r-- | 4.5 KB |
errors.rst | -rw-r--r-- | 410 bytes |
faq.rst | -rw-r--r-- | 15.2 KB |
glossary.rst | -rw-r--r-- | 4.0 KB |
gpu_memory_allocation.rst | -rw-r--r-- | 2.8 KB |
index.rst | -rw-r--r-- | 1.7 KB |
jax.dlpack.rst | -rw-r--r-- | 100 bytes |
jax.experimental.host_callback.rst | -rw-r--r-- | 281 bytes |
jax.experimental.loops.rst | -rw-r--r-- | 136 bytes |
jax.experimental.maps.rst | -rw-r--r-- | 153 bytes |
jax.experimental.optimizers.rst | -rw-r--r-- | 172 bytes |
jax.experimental.pjit.rst | -rw-r--r-- | 130 bytes |
jax.experimental.rst | -rw-r--r-- | 460 bytes |
jax.experimental.sparse.rst | -rw-r--r-- | 160 bytes |
jax.experimental.stax.rst | -rw-r--r-- | 154 bytes |
jax.flatten_util.rst | -rw-r--r-- | 218 bytes |
jax.image.rst | -rw-r--r-- | 233 bytes |
jax.lax.rst | -rw-r--r-- | 3.4 KB |
jax.lib.rst | -rw-r--r-- | 684 bytes |
jax.nn.initializers.rst | -rw-r--r-- | 488 bytes |
jax.nn.rst | -rw-r--r-- | 589 bytes |
jax.numpy.rst | -rw-r--r-- | 7.4 KB |
jax.ops.rst | -rw-r--r-- | 2.3 KB |
jax.profiler.rst | -rw-r--r-- | 785 bytes |
jax.random.rst | -rw-r--r-- | 819 bytes |
jax.rst | -rw-r--r-- | 2.2 KB |
jax.scipy.rst | -rw-r--r-- | 4.0 KB |
jax.tree_util.rst | -rw-r--r-- | 465 bytes |
jax_internal_api.rst | -rw-r--r-- | 161 bytes |
jaxpr.rst | -rw-r--r-- | 17.0 KB |
multi_process.md | -rw-r--r-- | 6.3 KB |
profiling.md | -rw-r--r-- | 10.0 KB |
pytrees.md | -rw-r--r-- | 9.0 KB |
rank_promotion_warning.rst | -rw-r--r-- | 1.6 KB |
requirements.txt | -rw-r--r-- | 404 bytes |
transformations.md | -rw-r--r-- | 8.6 KB |
type_promotion.rst | -rw-r--r-- | 12.3 KB |
Computing file changes ...