jax-cuda13-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

Apache-2.0 13 个版本
JAX team <jax-dev@google.com>
主页
Homepage
安装
pip install jax-cuda13-pjrt
poetry add jax-cuda13-pjrt
pipenv install jax-cuda13-pjrt
conda install jax-cuda13-pjrt
分类
Development Status :: 5 - Production/Stable Programming Language :: Python :: 3 Programming Language :: Python :: Free Threading :: 3 - Stable
版本列表
0.10.2 2026-06-17
0.10.1 2026-05-20
0.10.0 2026-04-16
0.9.2 2026-03-18
0.9.1 2026-03-02
0.9.0.1 2026-02-05
0.9.0 2026-01-20
0.8.3 2026-01-29
0.8.2 2025-12-18
0.8.1 2025-11-18
0.8.0 2025-10-15
0.7.2 2025-09-16
0.0.1rc0 2025-07-15