15 best Python GPU libraries in 2023

share link

by marketing.admin@openweaver.com dot icon Updated: Jan 15, 2023

technology logo
technology logo

Guide Kit Guide Kit  

Python is the most popular programming language in the world. Its success lies in its versatility, allowing developers to create everything from simple APIs to complex applications. For machine learning and deep learning, Python has become a preferred language because of its flexibility. The data science and machine learning community has been developing many open source libraries for Python. GPUs are highly specialized chips designed to perform matrix multiplication operations at blazing speeds. Although they were initially intended for rendering computer graphics on screens, they have proved quite useful for machine learning applications as well. Python has a number of libraries that make it easy for us to leverage GPUs for both training and inference tasks. Some of these focus on improving generic performance by leveraging CUDA primitives and it provide higher level abstractions that allow you to quickly build complex architectures without worrying about implementation details. Some of the most popular open-source libraries for Python GPU among developers are: Jax - Composable transformations of Python NumPy programs; kitty - Cross platform, fast, feature rich, GPU based terminal; Image AI - python library built to empower developers to build applications.

jaxby google

Python doticonstar image 23518 doticonVersion:jax-v0.4.12doticon
License: Permissive (Apache-2.0)

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Support
    Quality
      Security
        License
          Reuse

            jaxby google

            Python doticon star image 23518 doticonVersion:jax-v0.4.12doticon License: Permissive (Apache-2.0)

            Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
            Support
              Quality
                Security
                  License
                    Reuse

                      kittyby kovidgoyal

                      Python doticonstar image 18937 doticonVersion:nightlydoticon
                      License: Strong Copyleft (GPL-3.0)

                      Cross-platform, fast, feature-rich, GPU based terminal

                      Support
                        Quality
                          Security
                            License
                              Reuse

                                kittyby kovidgoyal

                                Python doticon star image 18937 doticonVersion:nightlydoticon License: Strong Copyleft (GPL-3.0)

                                Cross-platform, fast, feature-rich, GPU based terminal
                                Support
                                  Quality
                                    Security
                                      License
                                        Reuse

                                          tvmby apache

                                          Python doticonstar image 9868 doticonVersion:v0.12.0doticon
                                          License: Permissive (Apache-2.0)

                                          Open deep learning compiler stack for cpu, gpu and specialized accelerators

                                          Support
                                            Quality
                                              Security
                                                License
                                                  Reuse

                                                    tvmby apache

                                                    Python doticon star image 9868 doticonVersion:v0.12.0doticon License: Permissive (Apache-2.0)

                                                    Open deep learning compiler stack for cpu, gpu and specialized accelerators
                                                    Support
                                                      Quality
                                                        Security
                                                          License
                                                            Reuse

                                                              ImageAIby OlafenwaMoses

                                                              Python doticonstar image 7811 doticonVersion:test-resources-v3doticon
                                                              License: Permissive (MIT)

                                                              A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities

                                                              Support
                                                                Quality
                                                                  Security
                                                                    License
                                                                      Reuse

                                                                        ImageAIby OlafenwaMoses

                                                                        Python doticon star image 7811 doticonVersion:test-resources-v3doticon License: Permissive (MIT)

                                                                        A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
                                                                        Support
                                                                          Quality
                                                                            Security
                                                                              License
                                                                                Reuse

                                                                                  cupyby cupy

                                                                                  Python doticonstar image 6962 doticonVersion:v12.1.0.post1doticon
                                                                                  License: Permissive (MIT)

                                                                                  NumPy & SciPy for GPU

                                                                                  Support
                                                                                    Quality
                                                                                      Security
                                                                                        License
                                                                                          Reuse

                                                                                            cupyby cupy

                                                                                            Python doticon star image 6962 doticonVersion:v12.1.0.post1doticon License: Permissive (MIT)

                                                                                            NumPy & SciPy for GPU
                                                                                            Support
                                                                                              Quality
                                                                                                Security
                                                                                                  License
                                                                                                    Reuse

                                                                                                      AlphaPoseby MVIG-SJTU

                                                                                                      Python doticonstar image 7071 doticonVersion:Currentdoticon
                                                                                                      License: Others (Non-SPDX)

                                                                                                      Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

                                                                                                      Support
                                                                                                        Quality
                                                                                                          Security
                                                                                                            License
                                                                                                              Reuse

                                                                                                                AlphaPoseby MVIG-SJTU

                                                                                                                Python doticon star image 7071 doticonVersion:Currentdoticon License: Others (Non-SPDX)

                                                                                                                Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
                                                                                                                Support
                                                                                                                  Quality
                                                                                                                    Security
                                                                                                                      License
                                                                                                                        Reuse

                                                                                                                          incubator-tvmby apache

                                                                                                                          Python doticonstar image 5710 doticonVersion:v0.6.1doticon
                                                                                                                          License: Permissive (Apache-2.0)

                                                                                                                          Open deep learning compiler stack for cpu, gpu and specialized accelerators

                                                                                                                          Support
                                                                                                                            Quality
                                                                                                                              Security
                                                                                                                                License
                                                                                                                                  Reuse

                                                                                                                                    incubator-tvmby apache

                                                                                                                                    Python doticon star image 5710 doticonVersion:v0.6.1doticon License: Permissive (Apache-2.0)

                                                                                                                                    Open deep learning compiler stack for cpu, gpu and specialized accelerators
                                                                                                                                    Support
                                                                                                                                      Quality
                                                                                                                                        Security
                                                                                                                                          License
                                                                                                                                            Reuse

                                                                                                                                              AidLearning-FrameWorkby aidlearning

                                                                                                                                              Python doticonstar image 5166 doticonVersion:v1.3.0doticon
                                                                                                                                              License: Others (Non-SPDX)

                                                                                                                                              🔥🔥🔥AidLearning is a powerful AIOT development platform, AidLearning builds a linux env supporting GUI, deep learning and visual IDE on Android...Now Aid supports CPU+GPU+NPU for inference with high performance acceleration...Linux on Android or HarmonyOS

                                                                                                                                              Support
                                                                                                                                                Quality
                                                                                                                                                  Security
                                                                                                                                                    License
                                                                                                                                                      Reuse

                                                                                                                                                        AidLearning-FrameWorkby aidlearning

                                                                                                                                                        Python doticon star image 5166 doticonVersion:v1.3.0doticon License: Others (Non-SPDX)

                                                                                                                                                        🔥🔥🔥AidLearning is a powerful AIOT development platform, AidLearning builds a linux env supporting GUI, deep learning and visual IDE on Android...Now Aid supports CPU+GPU+NPU for inference with high performance acceleration...Linux on Android or HarmonyOS
                                                                                                                                                        Support
                                                                                                                                                          Quality
                                                                                                                                                            Security
                                                                                                                                                              License
                                                                                                                                                                Reuse

                                                                                                                                                                  gpuwebby gpuweb

                                                                                                                                                                  Python doticonstar image 2949 doticonVersion:Currentdoticon
                                                                                                                                                                  License: Others (Non-SPDX)

                                                                                                                                                                  Where the GPU for the Web work happens!

                                                                                                                                                                  Support
                                                                                                                                                                    Quality
                                                                                                                                                                      Security
                                                                                                                                                                        License
                                                                                                                                                                          Reuse

                                                                                                                                                                            gpuwebby gpuweb

                                                                                                                                                                            Python doticon star image 2949 doticonVersion:Currentdoticon License: Others (Non-SPDX)

                                                                                                                                                                            Where the GPU for the Web work happens!
                                                                                                                                                                            Support
                                                                                                                                                                              Quality
                                                                                                                                                                                Security
                                                                                                                                                                                  License
                                                                                                                                                                                    Reuse
                                                                                                                                                                                      Python doticonstar image 3918 doticonVersion:v0.0.1-dev9doticon
                                                                                                                                                                                      License: Permissive (Apache-2.0)

                                                                                                                                                                                      High-performance TensorFlow library for quantitative finance.

                                                                                                                                                                                      Support
                                                                                                                                                                                        Quality
                                                                                                                                                                                          Security
                                                                                                                                                                                            License
                                                                                                                                                                                              Reuse

                                                                                                                                                                                                tf-quant-financeby google

                                                                                                                                                                                                Python doticon star image 3918 doticonVersion:v0.0.1-dev9doticon License: Permissive (Apache-2.0)

                                                                                                                                                                                                High-performance TensorFlow library for quantitative finance.
                                                                                                                                                                                                Support
                                                                                                                                                                                                  Quality
                                                                                                                                                                                                    Security
                                                                                                                                                                                                      License
                                                                                                                                                                                                        Reuse
                                                                                                                                                                                                          Python doticonstar image 315 doticonVersion:v3.1doticon
                                                                                                                                                                                                          License: Permissive (Apache-2.0)

                                                                                                                                                                                                          This is a repository for an object detection inference API using the Tensorflow framework.

                                                                                                                                                                                                          Support
                                                                                                                                                                                                            Quality
                                                                                                                                                                                                              Security
                                                                                                                                                                                                                License
                                                                                                                                                                                                                  Reuse

                                                                                                                                                                                                                    BMW-TensorFlow-Inference-API-GPUby BMW-InnovationLab

                                                                                                                                                                                                                    Python doticon star image 315 doticonVersion:v3.1doticon License: Permissive (Apache-2.0)

                                                                                                                                                                                                                    This is a repository for an object detection inference API using the Tensorflow framework.
                                                                                                                                                                                                                    Support
                                                                                                                                                                                                                      Quality
                                                                                                                                                                                                                        Security
                                                                                                                                                                                                                          License
                                                                                                                                                                                                                            Reuse

                                                                                                                                                                                                                              ESRNN-GPUby damitkwr

                                                                                                                                                                                                                              Python doticonstar image 272 doticonVersion:Currentdoticon
                                                                                                                                                                                                                              License: Permissive (MIT)

                                                                                                                                                                                                                              PyTorch GPU implementation of the ES-RNN model for time series forecasting

                                                                                                                                                                                                                              Support
                                                                                                                                                                                                                                Quality
                                                                                                                                                                                                                                  Security
                                                                                                                                                                                                                                    License
                                                                                                                                                                                                                                      Reuse

                                                                                                                                                                                                                                        ESRNN-GPUby damitkwr

                                                                                                                                                                                                                                        Python doticon star image 272 doticonVersion:Currentdoticon License: Permissive (MIT)

                                                                                                                                                                                                                                        PyTorch GPU implementation of the ES-RNN model for time series forecasting
                                                                                                                                                                                                                                        Support
                                                                                                                                                                                                                                          Quality
                                                                                                                                                                                                                                            Security
                                                                                                                                                                                                                                              License
                                                                                                                                                                                                                                                Reuse

                                                                                                                                                                                                                                                  BMW-YOLOv4-Inference-API-GPUby BMW-InnovationLab

                                                                                                                                                                                                                                                  Python doticonstar image 249 doticonVersion:v3.1doticon
                                                                                                                                                                                                                                                  License: Permissive (BSD-3-Clause)

                                                                                                                                                                                                                                                  This is a repository for an nocode object detection inference API using the Yolov3 and Yolov4 Darknet framework.

                                                                                                                                                                                                                                                  Support
                                                                                                                                                                                                                                                    Quality
                                                                                                                                                                                                                                                      Security
                                                                                                                                                                                                                                                        License
                                                                                                                                                                                                                                                          Reuse

                                                                                                                                                                                                                                                            BMW-YOLOv4-Inference-API-GPUby BMW-InnovationLab

                                                                                                                                                                                                                                                            Python doticon star image 249 doticonVersion:v3.1doticon License: Permissive (BSD-3-Clause)

                                                                                                                                                                                                                                                            This is a repository for an nocode object detection inference API using the Yolov3 and Yolov4 Darknet framework.
                                                                                                                                                                                                                                                            Support
                                                                                                                                                                                                                                                              Quality
                                                                                                                                                                                                                                                                Security
                                                                                                                                                                                                                                                                  License
                                                                                                                                                                                                                                                                    Reuse

                                                                                                                                                                                                                                                                      bert-multi-gpuby HaoyuHu

                                                                                                                                                                                                                                                                      Python doticonstar image 176 doticonVersion:v1.1.0doticon
                                                                                                                                                                                                                                                                      License: Permissive (Apache-2.0)

                                                                                                                                                                                                                                                                      Feel free to fine tune large BERT models with Multi-GPU and FP16 support.

                                                                                                                                                                                                                                                                      Support
                                                                                                                                                                                                                                                                        Quality
                                                                                                                                                                                                                                                                          Security
                                                                                                                                                                                                                                                                            License
                                                                                                                                                                                                                                                                              Reuse

                                                                                                                                                                                                                                                                                bert-multi-gpuby HaoyuHu

                                                                                                                                                                                                                                                                                Python doticon star image 176 doticonVersion:v1.1.0doticon License: Permissive (Apache-2.0)

                                                                                                                                                                                                                                                                                Feel free to fine tune large BERT models with Multi-GPU and FP16 support.
                                                                                                                                                                                                                                                                                Support
                                                                                                                                                                                                                                                                                  Quality
                                                                                                                                                                                                                                                                                    Security
                                                                                                                                                                                                                                                                                      License
                                                                                                                                                                                                                                                                                        Reuse
                                                                                                                                                                                                                                                                                          Python doticonstar image 120 doticonVersion:Currentdoticon
                                                                                                                                                                                                                                                                                          License: Permissive (MIT)

                                                                                                                                                                                                                                                                                          Hands-On GPU Programming with Python and CUDA, published by Packt

                                                                                                                                                                                                                                                                                          Support
                                                                                                                                                                                                                                                                                            Quality
                                                                                                                                                                                                                                                                                              Security
                                                                                                                                                                                                                                                                                                License
                                                                                                                                                                                                                                                                                                  Reuse

                                                                                                                                                                                                                                                                                                    Hands-On-GPU-Programming-with-Python-and-CUDAby PacktPublishing

                                                                                                                                                                                                                                                                                                    Python doticon star image 120 doticonVersion:Currentdoticon License: Permissive (MIT)

                                                                                                                                                                                                                                                                                                    Hands-On GPU Programming with Python and CUDA, published by Packt
                                                                                                                                                                                                                                                                                                    Support
                                                                                                                                                                                                                                                                                                      Quality
                                                                                                                                                                                                                                                                                                        Security
                                                                                                                                                                                                                                                                                                          License
                                                                                                                                                                                                                                                                                                            Reuse

                                                                                                                                                                                                                                                                                                              See similar Kits and Libraries