PyRank

Gpu Python Packages

Python packages with the GitHub topic gpu. Sorted by relevance, with stars and monthly downloads.
pytorch
torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

93.7M 101K 28K
NVIDIA
nvidia-nccl-cu12

Optimized primitives for collective multi-GPU communication

50.8M 5K 1K
apache
apache-tvm-ffi

Open ABI and FFI for Machine Learning Systems

6.6M 427 83
catboost
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.

6.3M 9K 1K
NVIDIA
nvidia-cutlass-dsl

CUDA Templates and Python DSLs for High-Performance Linear Algebra

4.7M 10K 2K
NVIDIA
nvidia-cutlass-dsl-libs-base

CUDA Templates and Python DSLs for High-Performance Linear Algebra

4.5M 10K 2K
flashinfer-ai
flashinfer-python

FlashInfer: Kernel Library for LLM Serving

4.5M 6K 1K
cupy
cupy-cuda12x

NumPy & SciPy for GPU

3.6M 12K 1K
NVIDIA
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.

3.4M 862 197
flashinfer-ai
flashinfer-cubin

FlashInfer: Kernel Library for LLM Serving

3M 6K 1K
nvidia
cuda-tile

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

3M 2K 140
ashvardanian
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 🦖

2.8M 4K 128
isl-org
open3d

Open3D: A Modern Library for 3D Data Processing

1.9M 14K 3K
NVIDIA
nvidia-cutlass-dsl-libs-cu13

CUDA Templates and Python DSLs for High-Performance Linear Algebra

1.7M 10K 2K
NVIDIA
warp-lang

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

1.5M 7K 554
deepspeedai
deepspeed

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

1.4M 43K 5K
runpod
runpod

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

1.3M 302 119
skypilot-org
skypilot

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

1.3M 10K 1K
PennyLaneAI
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
wookayin
gpustat

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

969K 4K 286
fastai
fastai

The fastai deep learning library

667K 28K 8K
Qiskit
qiskit-aer

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

641K 675 437
cupy
cupy-cuda13x

NumPy & SciPy for GPU

532K 12K 1K
clearml
clearml-agent

ClearML Agent - MLOps/LLMOps made easy. MLOps/LLMOps scheduler & orchestration solution

512K 306 118