Package profile
jax-cuda12-pjrt
- Summary: JAX XLA PJRT Plugin for NVIDIA GPUs
- Author: JAX team
- License: Apache-2.0
- Homepage: https://github.com/jax-ml/jax
- Source: https://github.com/jax-ml/jax
- Number of releases: 40
- First release: 0.4.23 on 2023-12-15
- Latest release: 0.12.0.1 on 2026-10-21
- Latest release size: 177.2 MB (wheel)