Gpu Computing Python Packages

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.

pycuda

CUDA integration for Python, plus shiny features

catboost-dev

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.

xsuite

Suite of python packages for multiparticle simulations of particle accelerators.

cuda-cccl

CUDA Core Compute Libraries

dpctl

Python SYCL bindings and SYCL-based Python Array API library

tff-nightly

High-performance TensorFlow library for quantitative finance.

pysdm

Pythonic particle-based (super-droplet) cloud microphysics (and aqueous-chemistry) package with box, parcel & 1D/2D prescribed-flow examples in Python, Julia and Matlab

emmy-ml

Optimized GPU compiler for LLM inference. Choose from a list of optimized recipes or optimize your own model via kernel fusion, autotuning, and advanced scheduling. Run benchmarks across different GPU types and configurations, track results and share experiments with the community.

pysdm-examples

Pythonic particle-based (super-droplet) cloud microphysics (and aqueous-chemistry) package with box, parcel & 1D/2D prescribed-flow examples in Python, Julia and Matlab

pyluxcore

LuxCore source repository

openhcs

Typed, reproducible high-content microscopy workflows across GUI, Python, CellProfiler, Napari/Fiji, GPU, and local MCP agents.

bindsnet

Simulation of spiking neural networks (SNNs) using PyTorch.

cthreads

Typed Python that compiles to C++ and runs on real OS threads. Same jobs/pools/sync feel as normal Python, without the GIL or multiprocessing overhead. Optional Vulkan @Gpu kernels via pip install cthreads-gpu

brian2genn

Brian 2 frontend to the GeNN simulator

nutens

Tensor based engine for calculating neutrino oscillation probabilities in a fast, flexible, and differentiable way

cthreads-gpu

Typed Python that compiles to C++ and runs on real OS threads. Same jobs/pools/sync feel as normal Python, without the GIL or multiprocessing overhead. Optional Vulkan @Gpu kernels via pip install cthreads-gpu

flimkit

Alternative FLIM fitting/phasor analysis for FLIM files

kernel-tuner

Kernel Tuner

oapython

GPU-first compute, ML, vision, audio, media, UI, plotting, and cryptography on Vulkan.

evomo

EvoMO is a GPU-accelerated library for evolutionary multiobjective optimization (EMO)

torch-sla

torch sparse linear algebra library

rgpu

Keep Python on your laptop. Run PyTorch operations and hold tensors on a remote GPU, including from a Mac with no CUDA installation.

tf-quant-finance

High-performance TensorFlow library for quantitative finance.