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:21:58 2025 |
| md5 checksum | 45f05f89599f93b0f3461ae4d5448d0d |
| arch | x86_64 |
| build | py311h06a4308_0 |
| constrains | protobuf >=3.13,<4 |
| depends | jaxlib >=0.4.19, ml_dtypes >=0.2.0, numpy >=1.23.2,<2.0a0, opt_einsum, python >=3.11,<3.12.0a0, scipy >=1.9 |
| license | Apache-2.0 |
| license_family | APACHE |
| md5 | 45f05f89599f93b0f3461ae4d5448d0d |
| name | jax |
| platform | linux |
| sha1 | ae603d8c6f4542ae5d6ce5dc7f7634dcbee864b7 |
| sha256 | e1fead3da0c88e8b6a9ac70523aaa2187a8f28dfff3c8dafc6d523f07c23b797 |
| size | 3883071 |
| subdir | linux-64 |
| timestamp | 1706213273639 |
| version | 0.4.23 |