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-cuda13x

NumPy & SciPy for GPU

cupy-cuda11x

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

iree-base-compiler

A retargetable MLIR-based machine learning compiler and runtime toolkit.

iree-base-runtime

A retargetable MLIR-based machine learning compiler and runtime toolkit.

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

cupy-rocm-7-0

NumPy & SciPy for GPU

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).

pennylane-lightning-gpu

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

iree-tools-tflite

A retargetable MLIR-based machine learning compiler and runtime toolkit.

iree-tools-tf

A retargetable MLIR-based machine learning compiler and runtime toolkit.

cupy-cuda102

NumPy & SciPy for GPU

iree-runtime

A retargetable MLIR-based machine learning compiler and runtime toolkit.

aphrodite-engine

Large-scale LLM inference engine

cupy-cuda100

NumPy & SciPy for GPU

cupy-cuda110

NumPy & SciPy for GPU

cupy-cuda101

NumPy & SciPy for GPU

cupy-cuda111

NumPy & SciPy for GPU

cupy-cuda90

NumPy & SciPy for GPU