jax-rocm7-pjrt

JAX XLA PJRT Plugin for AMD GPUs (ROCm:unknown)

Apache-2.0 16 个版本
Ruturaj4 <Ruturaj.Vaidya@amd.com>
主页
Homepage
安装
pip install jax-rocm7-pjrt
poetry add jax-rocm7-pjrt
pipenv install jax-rocm7-pjrt
conda install jax-rocm7-pjrt
分类
Development Status :: 3 - Alpha Programming Language :: Python :: 3
版本列表
0.10.1 2026-06-16
0.10.0 2026-05-12
0.10.0.post1 2026-06-16
0.9.2 2026-04-30
0.9.2.post1 2026-04-30
0.9.1 2026-03-13
0.9.1.post6 2026-06-11
0.9.1.post5 2026-05-13
0.9.1.post4 2026-04-22
0.9.1.post3 2026-03-23
0.9.1.post2 2026-03-18
0.9.1.post1 2026-03-13
0.9.2b1 2026-04-07
0.7.1 2025-11-05
0.6.0 2025-10-07
0.6.0rc1 2025-10-03