Rocm Python Packages

cupy-cuda12x

NumPy & SciPy for GPU

3.6M 12K 1K

pennylane-lightning

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

1.1M 140 55

cupy-cuda13x

NumPy & SciPy for GPU

532K 12K 1K

cupy-cuda11x

NumPy & SciPy for GPU

148K 12K 1K

lmcache

LMCache: Supercharge Your LLM with the Fastest KV Cache Layer

77K 10K 1K

cupy

NumPy & SciPy for GPU

49K 12K 1K

iree-base-compiler

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

44K 4K 945

conch-triton-kernels

A "standard library" of Triton kernels.

42K 26 3

iree-base-runtime

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

39K 4K 945

apache-tvm

Open Machine Learning Compiler Framework

20K 14K 4K

pennylane-lightning-kokkos

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

19K 140 55

cupy-rocm-7-0

NumPy & SciPy for GPU

15K 12K 1K

deepmd-kit

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

14K 2K 629

pennylane-lightning-gpu

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

11K 140 55

iree-tools-tflite

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

10K 4K 945

iree-tools-tf

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

10K 4K 945

cupy-cuda102

NumPy & SciPy for GPU

6K 12K 1K

cupy-cuda100

NumPy & SciPy for GPU

5K 12K 1K

cupy-cuda101

NumPy & SciPy for GPU

5K 12K 1K

iree-runtime

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

4K 4K 945

cupy-cuda110

NumPy & SciPy for GPU

4K 12K 1K

cupy-cuda92

NumPy & SciPy for GPU

4K 12K 1K

cupy-cuda90

NumPy & SciPy for GPU

4K 12K 1K

cupy-cuda111

NumPy & SciPy for GPU

4K 12K 1K