Python Cuda

Open-source Python projects categorized as Cuda

Top 23 Python Cuda Projects

  • Numba

    NumPy aware dynamic Python compiler using LLVM

    Project mention: Python Algotrading with Machine Learning | | 2023-05-30

    A super-fast backtesting engine built in NumPy and accelerated with Numba.

  • 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.

    Project mention: Boosting Algorithms | | 2022-07-05
  • InfluxDB

    Access the most powerful time series database as a service. Ingest, store, & analyze all types of time series data in a fully-managed, purpose-built database. Keep data forever with low-cost storage and superior data compression.

  • cupy

    NumPy & SciPy for GPU

    Project mention: Speed Up Your Physics Simulations (250x Faster Than NumPy) Using PyTorch. Episode 1: The Boltzmann Distribution | | 2023-04-13

    I'd also recommend checking out CuPy which aims to fully re-implement the Numpy api for CUDA GPUs, while taking advantage of Nvidia's specialized libraries like cuBLAS, cuRAND, cuSOLVER etc. The tradeoff being that it only works with Nvidia GPUs.

  • chainer

    A flexible framework of neural networks for deep learning

    Project mention: Protip: the upscaler matters a lot | | 2023-01-13

    Sorry maybe someone could chime in and help but I use chainer to upscale.

  • cog

    Containers for machine learning

    Project mention: Opinions on Cog: Containers for machine learning | | 2023-01-15

    Then I discovered Cog: Containers for Machine Learning. Looks like a way more flexible solution to plug in the existing infrastructure: you write your custom code and Cog plugs it in a Docker image with FastAPI, no extra ecosystem complexity added.

  • jittor

    Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

    Project mention: Jittor: High-performance deep learning framework based on JIT and meta-operators | | 2023-04-02
  • nvitop

    An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

    Project mention: Why does stable diffusion hold onto my vram even when it’s doing nothing. It works great for a few images and then it racks up so much vram usage it just won’t do anything anymore and errors out. Is there a way to free up VRAM every so often? RX 6700XT 12GB VRAM | | 2023-03-25

    Hey, thanks for that. I think I found what you mentioned:

  • Sonar

    Write Clean Python Code. Always.. Sonar helps you commit clean code every time. With over 225 unique rules to find Python bugs, code smells & vulnerabilities, Sonar finds the issues while you focus on the work.

  • PyCUDA

    CUDA integration for Python, plus shiny features

  • torchrec

    Pytorch domain library for recommendation systems

  • 3d-ken-burns

    an implementation of 3D Ken Burns Effect from a Single Image using PyTorch

    Project mention: Making a trailer for my book with midjourney. What do you think? | | 2023-04-16

    I guess you could use a free video editor (Davinci resolve is free with less features than payed version). And then try to use this open source script:, but it would definitely be harder.

  • QualityScaler

    QualityScaler - image/video deeplearning upscaling for any GPU

    Project mention: QualityScaler 2.1 - image/video AI upscaler | | 2023-05-15
  • viseron

    Self-hosted, local only NVR and AI Computer Vision software. With features such as object detection, motion detection, face recognition and more, it gives you the power to keep an eye on your home, office or any other place you want to monitor.

    Project mention: Viseron 2.2.0 - Self-hosted, local only NVR and AI Computer Vision software. | | 2023-04-22
  • pyopencl

    OpenCL integration for Python, plus shiny features

    Project mention: An example for OpenCL 3.0? | | 2023-03-11

    Please note that OpenCL consists of two parts: host API and a separate language which is used to write kernels (code which is going to be offloaded to devices). OpenCL specification describes host APIs as C-style APIs and that is what implementors has to provide. However, there are number of various libraries which provides bindings for other languages: - C++ - Python - Go - Rust

  • tsdf-fusion-python

    Python code to fuse multiple RGB-D images into a TSDF voxel volume.

  • scikit-cuda

    Python interface to GPU-powered libraries

  • caer

    High-performance Vision library in Python. Scale your research, not boilerplate.

  • TransformerEngine

    A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in both training and inference.

    Project mention: Benchmarking Large Language Models on NVIDIA H100 GPUs with CoreWeave (Part 1) | | 2023-04-30

    4090 now has its 8-bit float enabled as well, see the [transformer engine issue](

  • FastFold

    Optimizing AlphaFold Training and Inference on GPU Clusters (by hpcaitech)

  • arrayfire-python

    Python bindings for ArrayFire: A general purpose GPU library.

  • Pine

    :evergreen_tree: Aimbot powered by real-time object detection with neural networks, GPU accelerated with Nvidia. Optimized for use with CS:GO. (by petercunha)

  • plotoptix

    Data visualisation and ray tracing in Python based on OptiX 7.6 framework.

    Project mention: GitHub - rnd-team-dev/plotoptix: Data visualisation and ray tracing in Python based on OptiX 7.5 framework. | | 2022-10-30
  • dace

    DaCe - Data Centric Parallel Programming

    Project mention: FPGA high-level programming | | 2023-02-02

    I had this recommended to me, but have yet to try it out:

  • warp-drive

    Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)

  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2023-05-30.

Python Cuda related posts


What are some of the best open-source Cuda projects in Python? This list will help you:

Project Stars
1 Numba 8,633
2 catboost 7,160
3 cupy 6,924
4 chainer 5,802
5 cog 4,805
6 jittor 2,811
7 nvitop 1,950
8 PyCUDA 1,547
9 torchrec 1,437
10 3d-ken-burns 1,427
11 QualityScaler 1,167
12 viseron 975
13 pyopencl 974
14 tsdf-fusion-python 954
15 scikit-cuda 930
16 caer 691
17 TransformerEngine 664
18 FastFold 475
19 arrayfire-python 406
20 Pine 396
21 plotoptix 392
22 dace 376
23 warp-drive 373
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives