kandi background
Explore Kits

15 best Python GPU libraries in 2022

by Open Weaver kits ✔ Updated: May 13, 2022

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 star image 17239 Version:jax-v0.3.6

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 star image 17239 Version:jax-v0.3.6 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 star image 13981 Version:nightly

License: Strong Copyleft (GPL-3.0)

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

Support
Quality
Security
License
Reuse

kittyby kovidgoyal

Python star image 13981 Version:nightly License: Strong Copyleft (GPL-3.0)

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

tvmby apache

Python star image 7990 Version:v0.8.0

License: Permissive (Apache-2.0)

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

Support
Quality
Security
License
Reuse

tvmby apache

Python star image 7990 Version:v0.8.0 License: Permissive (Apache-2.0)

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

ImageAIby OlafenwaMoses

Python star image 6818 Version:essentials-v5

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 star image 6818 Version:essentials-v5 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 star image 5918 Version:v10.3.1

License: Permissive (MIT)

NumPy & SciPy for GPU

Support
Quality
Security
License
Reuse

cupyby cupy

Python star image 5918 Version:v10.3.1 License: Permissive (MIT)

NumPy & SciPy for GPU
Support
Quality
Security
License
Reuse

AlphaPoseby MVIG-SJTU

Python star image 5803 Version:Current

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 star image 5803 Version:Current 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 star image 5710 Version:v0.6.1

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 star image 5710 Version:v0.6.1 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 star image 4468 Version:v0.92

License: Others (Non-SPDX)

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

Support
Quality
Security
License
Reuse

AidLearning-FrameWorkby aidlearning

Python star image 4468 Version:v0.92 License: Others (Non-SPDX)

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

gpuwebby gpuweb

Python star image 2949 Version:Current

License: Others (Non-SPDX)

Where the GPU for the Web work happens!

Support
Quality
Security
License
Reuse

gpuwebby gpuweb

Python star image 2949 Version:Current License: Others (Non-SPDX)

Where the GPU for the Web work happens!
Support
Quality
Security
License
Reuse

tf-quant-financeby google

Python star image 2674 Version:v0.0.1-dev9

License: Permissive (Apache-2.0)

High-performance TensorFlow library for quantitative finance.

Support
Quality
Security
License
Reuse

tf-quant-financeby google

Python star image 2674 Version:v0.0.1-dev9 License: Permissive (Apache-2.0)

High-performance TensorFlow library for quantitative finance.
Support
Quality
Security
License
Reuse

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

Python star image 287 Version:v3.1

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 star image 287 Version:v3.1 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 star image 272 Version:Current

License: Permissive (MIT)

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

Support
Quality
Security
License
Reuse

ESRNN-GPUby damitkwr

Python star image 272 Version:Current 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 star image 249 Version:v3.1

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 star image 249 Version:v3.1 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 star image 176 Version:v1.1.0

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 star image 176 Version:v1.1.0 License: Permissive (Apache-2.0)

Feel free to fine tune large BERT models with Multi-GPU and FP16 support.
Support
Quality
Security
License
Reuse

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

Python star image 120 Version:Current

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 star image 120 Version:Current License: Permissive (MIT)

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

See similar Kits and Libraries

GPU
Hardware
  • © 2022 Open Weaver Inc.