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

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.