Gpu Programming Python Packages

scalene

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals

taichi

Productive, portable, and performant GPU programming in Python.

fragmentcolor

Universal GPU abstraction for Rust, Javascript, Python, Swift, and Kotlin.

geomstats

Computations and statistics on manifolds with geometric structures.

cuda-cccl

CUDA Core Compute Libraries

triton-transformer

Implementation of a Transformer, but completely in Triton

taichi-nightly

Productive, portable, and performant GPU programming in Python.

nabla-ml

Nabla: High-Performance Scientific Computing

mm-ptx

PTX Inject and Stack PTX for Python

greyhound-kernels

CuTe DSL kernels for PyTorch training and inference workloads

gpu-wait

A package to run commands when GPU resources are available

chronos-gpu

Chronos is a time-based GPU partitioning utility that allows multiple users or applications to share a single GPU by creating exclusive time-limited partitions with automatic expiration. Built with OpenCL, it works across platforms including macOS (Apple Silicon & Intel), Linux, and Windows.

numba-timer

A helper package to easily time Numba CUDA GPU events ⌛

stgraph

🌟 Compiler for vertex-centric programming of GNNs/TGNNs

cuda-parallel

cuda.parallel: (experimental) CUDA parallel algorithms for Python

cuda-cooperative

cuda.cooperative: (experimental) CUDA cooperative algorithms for Python