jax
Description
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
More information
- Homepage: https://pypi.python.org/pypi/jax
Available Versions of jax
Version | Module | Available on |
---|---|---|
0.3.9-foss-2021a-CUDA-11.3.1 | tools/jax/0.3.9-foss-2021a-CUDA-11.3.1 | Noctua 2 |
0.3.9-foss-2021a | tools/jax/0.3.9-foss-2021a | Noctua 1, Noctua 2 |
This table is generated automatically. If you need other versions please click pc2-support@uni-paderborn.de.
Usage Hints for jax
If you need support in using this software or example job scripts please click pc2-support@uni-paderborn.de.