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.

6.3M 9K 1K

pycuda

CUDA integration for Python, plus shiny features

88K 2K 299

catboost-dev

66K 9K 1K

xsuite

Suite of python packages for multiparticle simulations of particle accelerators.

42K 50 37

xdeps

Python data dependency manager

27K 2 15

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

19K 90 54

dpctl

Python SYCL bindings and SYCL-based Python Array API library

15K 127 38

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

13K 90 54

cuda-cccl

CUDA Core Compute Libraries

11K 2K 422

tff-nightly

High-performance TensorFlow library for quantitative finance.

9K 5K 691

pyluxcore

LuxCore source repository

7K 1K 165

xobjects

Library to create and manipulate serialized objects in CPU and GPU memory and to compile code on these platforms.

7K 4 22

xpart

Python package for generation and analysis of particles distributions

6K 4 28

kernel-tuner

Kernel Tuner

2K 403 67

torch-sla

torch sparse linear algebra library

2K 49 2

brian2genn

Brian 2 frontend to the GeNN simulator

1K 50 17

xtructure

JAX-optimized data structures

1K 23 0

m2m-vector-search

M2M (Machine-to-Memory) - High-performance vector database with GPU acceleration and energy-based features

1K 24 8

jaxamg

GPU-accelerated differentiable sparse linear solver library for JAX

984 32 1

vision-xformer

Vision Xformers

981 23 7

pandadock

PandaDock: Physics based Molecular Docking with GNN Scoring

881 98 18

maddening

Modular Automatic Differentiation and Data-Enhanced Neural-network INteracting Graph (MADDENING). A pure JAX end-to-end autodifferentiable graph based framework for multiphysics simulation.

859 1 0

nutens

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

807 1 0

exospherehost

Runtime for building and managing AI agents and Workflows. Easy to learn, fast to build, High Performance, Reliable by design, Intuitive UI, Production Ready.

799 200 43