Top 23 GPU Open-Source Projects

  • GitHub repo Pytorch

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Project mention: How do you multiply two sparse matrices? | reddit.com/r/pytorch | 2021-07-29

    You’ll save the memory, but it’ll be slower obviously. I’m not sure how performance critical this application is. This type of multiplication looks like an open issue still: https://github.com/pytorch/pytorch/issues/3158

  • GitHub repo alacritty

    A cross-platform, OpenGL terminal emulator.

    Project mention: Vim renders slowly the highlight/rest of buffer even with small files, what to do? | reddit.com/r/vim | 2021-07-25

    Try using Alacritty. I’m getting similar performance to MacVim on it.

  • GitHub repo fastai

    The fastai deep learning library

    Project mention: D I Refuse To Use Pytorch Because Its A Facebook | reddit.com/r/MachineLearning | 2020-12-29

    Also, not a single docstring to document any code in the library - https://github.com/fastai/fastai/blob/master/fastai/vision/learner.py

  • GitHub repo nvidia-docker

    Build and run Docker containers leveraging NVIDIA GPUs

    Project mention: Docker 1:20.10.6-1 Nvidia = Failed to initialize NVML: Unknown Error | reddit.com/r/archlinux | 2021-06-03
  • GitHub repo gpu.js

    GPU Accelerated JavaScript

    Project mention: For what projects, Nodejs is an absolute No No? | reddit.com/r/node | 2021-03-02
  • GitHub repo Deeplearning4j

    Model import deployment framework for retraining models (pytorch, tensorflow,keras) deploying in JVM Micro service environments, mobile devices, iot, and Apache Spark

    Project mention: Does Java has similar project like this one in C#? (ml, data) | reddit.com/r/java | 2021-05-23

    Also, the website is now redirected to: https://deeplearning4j.konduit.ai/

  • GitHub repo ImageAI

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

    Project mention: Robot Systems: Software Stack Overview | dev.to | 2021-04-04

    Image AI

  • GitHub repo 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: CatBoost Quickstart — ML Classification | dev.to | 2021-03-15

    CatBoost is an open source algorithm based on gradient boosted decision trees. It supports numerical, categorical and text features. Check out the docs.

  • GitHub repo H2O

    H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

  • GitHub repo AlphaPose

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

    Project mention: [P] Gait Recognition in the wild | reddit.com/r/MachineLearning | 2021-07-06

    The model is an ST-GCN trained in a completely unsupervised manner, from a LOT of skeleton sequences. Pose estimation was performed with AlphaPose, and tracked using SORT.

  • GitHub repo DeepSpeed

    DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

    Project mention: DeepSpeed: Deep Learning Optimization Library | news.ycombinator.com | 2021-06-10
  • GitHub repo cupy

    NumPy & SciPy for GPU

    Project mention: CuPy (NumPy/SciPy-Compatible Array Library for GPU-Accelerated Computing) | news.ycombinator.com | 2021-06-18
  • GitHub repo john

    John the Ripper jumbo - advanced offline password cracker, which supports hundreds of hash and cipher types, and runs on many operating systems, CPUs, GPUs, and even some FPGAs

    Project mention: Working on pulling the hash from a zip file to recover the password, zip2john is just dumping the file and giving me trash? | reddit.com/r/HowToHack | 2021-07-29

    Depending on the size of the files inside the zip it’s possible your large hash file is correct. See this link the BOM issue you’re having is an encoding issue. Convert the file from UTF8 BOM to just plain UTF8. The next issue you might run into is hashcat or John not liking the size of your hash file. I don’t have any good advice if you encounter that issue. ZIP hashes are known to be hard to work with.

  • GitHub repo stats

    macOS system monitor in your menu bar

    Project mention: Best menu bar apps? | reddit.com/r/macapps | 2021-07-27
  • GitHub repo gfx

    A low-overhead Vulkan-like GPU API for Rust.

    Project mention: Graphics Libraries? | reddit.com/r/rust_gamedev | 2021-06-03

    https://github.com/gfx-rs/gfx#hardware-abstraction-layer

  • GitHub repo Open3D

    Open3D: A Modern Library for 3D Data Processing

    Project mention: Show HN: 3dasd – open-source DIY room-scale 3D scanner | news.ycombinator.com | 2021-06-24

    Sweet work! If you are interested in 3D reconstruction, have a look at http://www.open3d.org.

  • GitHub repo Halide

    a language for fast, portable data-parallel computation

    Project mention: Image convolution optimisation strategies. | reddit.com/r/OpenCL | 2021-07-27

    You might want to create a kernel in Halide to check a reasonable tuned kernel: https://github.com/halide/Halide/blob/master/apps/blur/halide_blur_generator.cpp

  • GitHub repo scalene

    Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python

    Project mention: Measuring memory usage in Python: it’s tricky | news.ycombinator.com | 2021-07-05

    For folks looking for a good modern profiling tool for Python codebases: my current favourite is Scalene [0], which allows profiling CPU/GPU and memory usage in a performant way. The author gave a talk about it on last PyCon US [1].

    [0]: https://github.com/plasma-umass/scalene

  • GitHub repo cudf

    cuDF - GPU DataFrame Library

    Project mention: Looking for Suggestions for Testing Frameworks and Methodology | reddit.com/r/CUDA | 2021-06-15

    I don't know about MPI. But it works well with GPU. You need to write a custom timer to accurately measure GPU code runtime (example https://github.com/rapidsai/cudf/tree/branch-21.08/cpp/benchmarks/synchronization ).

  • GitHub repo pycaret

    An open-source, low-code machine learning library in Python

    Project mention: What is your DS stack? (and roast mine :) ) | reddit.com/r/datascience | 2021-04-06

    If you want to try pycaret exists, not sure how similar it is to caret, but it does all the steps in ML project. And Gluon for DL.

  • GitHub repo ArrayFire

    ArrayFire: a general purpose GPU library.

    Project mention: Arrayfire progressive performance decline? | reddit.com/r/rust | 2021-06-09

    Your Problem may be the lazy evaluation, see this issue: https://github.com/arrayfire/arrayfire/issues/1709

  • GitHub repo Ultralight

    Next-generation HTML renderer for apps and games

    Project mention: Hey Rustaceans! Got an easy question? Ask here (27/2021)! | reddit.com/r/rust | 2021-07-05

    I am looking for a library to integrate a HTML renderer for UI elements in my game. While searching for solutions I found https://ultralig.ht/ which exposes methods to render the browser view into a texture I can consume in my custom engine and upload to the GPU and render easily. However the Rust bindings have no useful documentation at all and the one and only example seems to create a specific window rather than using winit like I do.

  • GitHub repo Thrust

    The C++ parallel algorithms library.

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 2021-07-29.

Index

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

Project Stars
1 Pytorch 49,716
2 alacritty 33,815
3 fastai 21,191
4 nvidia-docker 13,415
5 gpu.js 12,662
6 Deeplearning4j 12,127
7 ImageAI 6,350
8 catboost 5,970
9 H2O 5,461
10 AlphaPose 5,325
11 DeepSpeed 5,278
12 cupy 5,241
13 john 5,136
14 stats 5,067
15 gfx 4,937
16 Open3D 4,860
17 Halide 4,463
18 scalene 4,258
19 cudf 4,071
20 pycaret 3,765
21 ArrayFire 3,559
22 Ultralight 3,423
23 Thrust 3,382