jax¶
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
homepage: https://pypi.python.org/pypi/jax
| version | versionsuffix | toolchain |
|---|---|---|
0.2.19 |
foss/2020b |
|
0.2.19 |
fosscuda/2020b |
|
0.2.20 |
foss/2021a |
|
0.2.24 |
-CUDA-11.3.1 |
foss/2021a |
0.2.24 |
foss/2021a |
|
0.3.14 |
-CUDA-11.7.0 |
foss/2022a |
0.3.23 |
-CUDA-11.4.1 |
foss/2021b |
0.3.23 |
foss/2022a |
|
0.3.25 |
-CUDA-11.7.0 |
foss/2022a |
0.3.25 |
foss/2022a |
|
0.3.9 |
-CUDA-11.3.1 |
foss/2021a |
0.3.9 |
foss/2021a |
|
0.4.25 |
-CUDA-12.1.1 |
gfbf/2023a |
0.4.25 |
gfbf/2023a |
|
0.4.4 |
-CUDA-11.7.0 |
foss/2022a |
0.4.4 |
foss/2022a |
(quick links: (all) - 0 - a - b - c - d - e - f - g - h - i - j - k - l - m - n - o - p - q - r - s - t - u - v - w - x - y - z)