jax-cuda12-plugin
JAX Plugin for NVIDIA GPUs
Installation
In a virtualenv (see these instructions if you need to create one):
pip3 install jax-cuda12-plugin
Releases
| Version | Released | Bullseye Python 3.9 |
Bookworm Python 3.11 |
Files |
|---|---|---|---|---|
| 0.5.3 | 2025-03-19 | |||
|
|
||||
| 0.5.1 | 2025-02-24 | |||
|
|
||||
| 0.5.0 | 2025-01-17 | |||
|
|
||||
| 0.4.38 | 2024-12-17 | |||
|
|
||||
| 0.4.36 | 2024-12-05 | |||
|
|
||||
| 0.4.35 | 2024-10-22 | |||
|
|
||||
| 0.4.34 | 2024-10-04 | |||
|
|
||||
| 0.4.33 | 2024-09-16 | |||
|
|
||||
| 0.4.32 yanked | 2024-09-11 | |||
|
|
||||
| 0.4.31 | 2024-07-30 | |||
|
|
||||
| 0.4.30 | 2024-06-18 | |||
|
|
||||
| 0.4.29 | 2024-06-10 | |||
|
|
||||
| 0.4.28 | 2024-05-09 | |||
|
|
||||
| 0.4.27 | 2024-05-07 | |||
|
|
||||
| 0.4.26 | 2024-04-03 | |||
|
|
||||
| 0.4.25 | 2024-02-26 | |||
|
|
||||
| 0.4.24 | 2024-02-07 | |||
|
|
||||
| 0.4.23 | 2023-12-14 | |||
|
|
||||
| 0.4.22 | 2023-12-14 | |||
|
|
||||
Issues with this package?
- Search issues for this package
- Package or version missing? Open a new issue
- Something else? Open a new issue