Rocm Python Packages

cupy-cuda12x

NumPy & SciPy for GPU

cupy-cuda13x

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.

lmcache

LMCache: Supercharge Your LLM with the Fastest KV Cache Layer

cupy

NumPy & SciPy for GPU

cupy-cuda11x

NumPy & SciPy for GPU

iree-base-compiler

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

deepmd-kit

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

iree-base-runtime

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

xllamacpp

xllamacpp - a Python wrapper of llama.cpp

conch-triton-kernels

A "standard library" of Triton kernels.

mojolearn

GPU machine learning in Mojo that supports bitwise identical inference and training on Apple Metal, NVIDIA CUDA and AMD HIP.

apache-tvm

Open Machine Learning Compiler Framework

iree-tools-tf

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

iree-tools-tflite

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.

cupy-rocm-7-0

NumPy & SciPy for GPU

mojolearn-nvidia

GPU machine learning in Mojo that supports bitwise identical inference and training on Apple Metal, NVIDIA CUDA and AMD HIP.

gafime

GPU-Accelerated Feature Interaction Mining Engine (GAFIME) — native heterogeneous execution across Rust CPU/SIMD, CUDA, ROCm/HIP, and Metal.

cupy-cuda102

NumPy & SciPy for GPU

cupy-cuda100

NumPy & SciPy for GPU

iree-runtime

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

pennylane-lightning-gpu

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

cupy-cuda90

NumPy & SciPy for GPU