Content License: Creative Commons Attribution 4.0 International (CC-BY-4.0)Credit must be given to the creatorDownloadsDownloadGPU Computing in PythonBen DuMay 9, 2020Things on this page are fragmentary and immature notes/thoughts of the author. Please read with your own judgement!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¶PyOpenGLMachine 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¶