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

107K 2K 298

xsuite

Suite of python packages for multiparticle simulations of particle accelerators.

84K 50 36

catboost-dev

49K 9K 1K

dpctl

Python SYCL bindings and SYCL-based Python Array API library

24K 124 36

tff-nightly

High-performance TensorFlow library for quantitative finance.

21K 5K 682

xdeps

Python data dependency manager

20K 2 14

cuda-cccl

CUDA Core Compute Libraries

11K 2K 391

pysdm

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

8K 86 53

pyluxcore

LuxCore source repository

7K 1K 164

xobjects

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

6K 4 21

pysdm-examples

PySDM usage examples reproducing results from literature and depicting how to use PySDM from Python Jupyter notebooks

6K 86 53

xpart

Python package for generation and analysis of particles distributions

6K 4 26

vision-xformer

Vision Xformers

2K 23 7

tf-quant-finance

High-performance TensorFlow library for quantitative finance.

1K 5K 682

brian2genn

An interface to use the GeNN framework as a device in Brian 2

1K 50 17

pycuda-gml

CUDA integration for Python, plus shiny features

1K 2K 298

cuda-cooperative

CUDA Core Compute Libraries

1K 2K 391

jqmc

jQMC code implements two real-space ab initio quantum Monte Carlo (QMC) methods. Variatioinal Monte Carlo (VMC) and lattice regularized diffusion Monte Carlo (LRDMC) methods. jQMC achieves high-performance computations especially on GPUs.

989 16 2

kernel-tuner

Kernel Tuner

957 395 67

liblaf-jarp

Python utilities for JAX and Warp with filtered jit, pytree helpers, and array-to-Warp conversion.

954 0 0

xmask

Collider configuration manager

953 0 8

torch-sla

torch sparse linear algebra library

940 34 1

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.

751 1 0