jax-rocm7-plugin

JAX Plugin for AMD GPUs (ROCm:unknown)

Apache-2.0 16 个版本 Python >=3.11
Ruturaj4 <Ruturaj.Vaidya@amd.com>
主页
Homepage
安装
pip install jax-rocm7-plugin
poetry add jax-rocm7-plugin
pipenv install jax-rocm7-plugin
conda install jax-rocm7-plugin
分类
Development Status :: 3 - Alpha Programming Language :: Python :: 3.11 Programming Language :: Python :: 3.12 Programming Language :: Python :: 3.13 Programming Language :: Python :: 3.14
版本列表
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.0 2026-03-09
0.9.2b1 2026-04-07
0.7.1 2025-11-05
0.6.0 2025-10-03