Gpu Python Packages

torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

catboost

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

apache-tvm-ffi

Open ABI and FFI for Machine Learning Systems

nvidia-cutlass-dsl-libs-base

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cudnn-frontend

cuDNN Frontend is NVIDIA's modern, open-source entry point to the cuDNN library and a growing collection of high-performance open-source kernels.

flashinfer-python

FlashInfer: Kernel Library for LLM Serving

nvidia-cutlass-dsl-libs-cu13

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl-libs-cu12

CUDA Templates and Python DSLs for High-Performance Linear Algebra

nvidia-cutlass-dsl-libs-core

CUDA Templates and Python DSLs for High-Performance Linear Algebra

stringzilla

Up to 100x faster strings for C, C++, CUDA, Python, Rust, Swift, JS, & Go, leveraging NEON, AVX2, AVX-512, SVE, GPGPU, & SWAR to accelerate search, hashing, sorting, edit distances, sketches, and memory ops 🦖

cuda-tile

cuTile is a programming model for writing parallel kernels for NVIDIA GPUs

nccl4py

Optimized primitives for collective multi-GPU communication

open3d

Open3D: A Modern Library for 3D Data Processing

cupy-cuda12x

NumPy & SciPy for GPU

warp-lang

A Python framework for GPU-accelerated simulation, robotics, and machine learning.

skypilot

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

gpustat

📊 A simple command-line utility for querying and monitoring GPU status

flashinfer-cubin

FlashInfer: Kernel Library for LLM Serving

deepspeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

pytorch-tokenizers

On-device AI across mobile, embedded and edge for PyTorch

runpod

🐍 | Python library for Runpod API and serverless worker SDK.

qiskit-aer

Aer is a high performance simulator for quantum circuits that includes noise models

cupy-cuda13x

NumPy & SciPy for GPU