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.

geomstats

Computations and statistics on manifolds with geometric structures.

cuda-cccl

CUDA Core Compute Libraries

fragmentcolor

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

taichi-nightly

Productive, portable, and performant GPU programming in Python.

triton-transformer

Implementation of a Transformer, but completely in Triton

cuda-cooperative

CUDA Core Compute Libraries

nabla-ml

Nabla: High-Performance Scientific Computing

mm-ptx

PTX Inject and Stack PTX for Python

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 Core Compute Libraries