Rocm Python Packages

cupy-cuda12x

NumPy & SciPy for GPU

pennylane-lightning

The Lightning plugin ecosystem provides fast quantum state-vector and tensor network simulators written in C++ for use with PennyLane.

cupy-cuda11x

NumPy & SciPy for GPU

cupy-cuda13x

NumPy & SciPy for GPU

lmcache

Supercharge Your LLM with the Fastest KV Cache Layer

conch-triton-kernels

A "standard library" of Triton kernels.

cupy

NumPy & SciPy for GPU

pennylane-lightning-kokkos

The Lightning plugin ecosystem provides fast quantum state-vector and tensor network simulators written in C++ for use with PennyLane.

deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics

pennylane-lightning-gpu

The Lightning plugin ecosystem provides fast quantum state-vector and tensor network simulators written in C++ for use with PennyLane.

isat-tuner

Inference Stack Auto-Tuner — one command to find the fastest ONNX inference config for any model on any GPU. Bayesian optimization, Pareto analysis, thermal-aware benchmarking, multi-provider (ROCm/CUDA/TensorRT/OpenVINO).

cupy-rocm-7-0

NumPy & SciPy for GPU

aphrodite-engine

Large-scale LLM inference engine

cupy-cuda102

NumPy & SciPy for GPU

cupy-rocm-5-0

NumPy & SciPy for GPU

cupy-cuda100

NumPy & SciPy for GPU

cupy-cuda101

NumPy & SciPy for GPU

cupy-rocm-4-3

NumPy & SciPy for GPU

cupy-cuda110

NumPy & SciPy for GPU

cupy-cuda90

NumPy & SciPy for GPU

cupy-cuda111

NumPy & SciPy for GPU

cupy-cuda92

NumPy & SciPy for GPU

cupy-cuda112

NumPy & SciPy for GPU

cupy-cuda91

NumPy & SciPy for GPU