General Purpose GPU Computing¶

pyculib¶

PyCUDA¶

CUDA integration for Python, plus shiny features.

cupy¶

NumPy-like API accelerated with CUDA.

cudf¶

cuDF - GPU DataFrame Library.

jax¶

cuml¶

Graphics Rendering¶

PyOpenGL

Machine Learning Libraries¶

Most machine learning libraries in Python support GPU computing. I list the most famous ones below.

LightGBM¶

AutoGluon¶

PyTorch¶

TensorFlow¶

Apache Ray¶

Horovod¶

ZeRO + DeepSpeed¶