jax-cuda13-plugin

JAX Plugin for NVIDIA GPUs

Apache-2.0 13 个版本 Python >=3.11
JAX team <jax-dev@google.com>
主页
Homepage
安装
pip install jax-cuda13-plugin
poetry add jax-cuda13-plugin
pipenv install jax-cuda13-plugin
conda install jax-cuda13-plugin
分类
Development Status :: 5 - Production/Stable Programming Language :: Python :: 3.11 Programming Language :: Python :: 3.12 Programming Language :: Python :: 3.13 Programming Language :: Python :: 3.14 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