JAX is Autograd and XLA, brought together for high-performance numerical computing and machine learning research. It provides composable transformations of Python+NumPy programs: differentiate, vectorize, parallelize, Just-In-Time compile to GPU/TPU, and more.
| Uploaded | Sun Mar 30 23:22:01 2025 |
| md5 checksum | 5a555ec8e35c9c790c402b0a64863364 |
| arch | x86_64 |
| build | py39h06a4308_0 |
| depends | importlib_metadata >=4.6, jaxlib >=0.4.14, ml_dtypes >=0.2.0, numpy >=1.22,<2.0a0, opt_einsum, python >=3.9,<3.10.0a0, scipy >=1.7 |
| license | Apache-2.0 |
| license_family | APACHE |
| md5 | 5a555ec8e35c9c790c402b0a64863364 |
| name | jax |
| platform | linux |
| sha1 | 025e0cb497969d13cd7c911627b772d45bc1f4de |
| sha256 | 3380933d34062483315cd151528436f6c3ba207b2bd3d8fdaaab7acb38313864 |
| size | 2730703 |
| subdir | linux-64 |
| timestamp | 1698071697986 |
| version | 0.4.16 |