Search 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.

cuda-cccl

CUDA Core Compute Libraries

geomstats

Computations and statistics on manifolds with geometric structures.

taichi-nightly

The Taichi Programming Language

fragmentcolor

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

cuda-cooperative

CUDA Core Compute Libraries

triton-transformer

Transformer in Triton

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.

stgraph

🌟 Compiler for vertex-centric programming of GNNs/TGNNs

numba-timer

A helper package to easily time Numba CUDA GPU events

cuda-parallel

CUDA Core Compute Libraries