Ross ROSS = Recommend OSS · open-source software intelligence for agents

Open-source gpu-computing software

544 open-source libraries, tools and applications tagged gpu-computing, ranked by how closely gpu-computing is their primary purpose, then by adoption and maintenance health. Each entry links to a page with sources, use cases and when to choose or avoid it.

544 products, primary matches first, then adoption-weighted; health v2 shown.

ProductHealth v2StarsMaturity
uber/aresdb
AresDB is a GPU-powered real-time analytics storage and query engine written in Go and C++ with CUDA kernels. It provides low query latency…
233078maintenance
ajbrock/BigGAN-PyTorch
A PyTorch implementation of BigGAN for large-scale high-fidelity image synthesis, written by the paper's first author. It supports multi-GP…
322928maintenance
Tencent/PhoenixGo
PhoenixGo is a Go (weiqi/baduk) AI engine from Tencent that implements the AlphaGo Zero approach of learning the game without human knowled…
232918maintenance
sxyu/svox2
Official PyTorch/CUDA implementation of Plenoxels, a method for optimizing sparse voxel radiance fields for novel view synthesis without ne…
322890maintenance
cybertronai/gradient-checkpointing
A Python library that reduces GPU memory usage when training very deep neural networks via gradient checkpointing, trading computation for …
322847maintenance
microsoft/pai
OpenPAI is an open-source AI platform from Microsoft that provides resource scheduling and cluster management for machine learning workload…
682700maintenance
unslothai/hyperlearn
HyperLearn is a Python library providing optimized reimplementations of classical machine learning and statistical algorithms (SVD, eigende…
322520maintenance
IST-DASLab/gptq
Reference implementation of GPTQ, a one-shot post-training weight quantization method for large generative transformer models based on appr…
322386maintenance
hunglc007/tensorflow-yolov4-tflite
A TensorFlow 2.x implementation of YOLOv4, YOLOv4-tiny, YOLOv3, and YOLOv3-tiny object detection models, with scripts that convert original…
322255maintenance
ashawkey/torch-ngp
A PyTorch implementation of instant-ngp's NeRF and SDF components with CUDA extensions and an interactive GUI for training and visualizatio…
322210maintenance
PyTorch-Encoding
A PyTorch computer vision toolkit with optimized CUDA backend, providing encoding layers, synchronized multi-GPU batch normalization, and p…
652045maintenance
darglein/ADOP
ADOP is a point-based differentiable neural rendering pipeline for scene refinement and novel view synthesis, implemented in C++/CUDA with …
232028maintenance
google/compare_gan
A TensorFlow library from Google Research providing configurable implementations of GAN components: losses, penalties, normalization techni…
101815maintenance
jkjung-avt/tensorrt_demos
A collection of Python demo programs showing how to optimize and run deep learning models (YOLOv4, YOLOv3, SSD, MTCNN, GoogLeNet, MODNet) w…
461793maintenance
Qihoo360/hbox
Hbox (formerly XLearning) is a scheduling platform that runs machine learning and deep learning frameworks like TensorFlow, MXNet, Caffe, a…
431728maintenance
uber-research/deep-neuroevolution
Distributed implementations of deep neuroevolution algorithms (ES, NS-ES, NSR-ES, DeepGA, Random Search) from Uber AI Labs research papers,…
321667maintenance
tensorflow/mesh
Mesh TensorFlow is a Python library and embedded language for specifying distributed tensor computations, letting users define how model di…
101629maintenance
maderix/ANE
A research project demonstrating backpropagation and neural network training directly on Apple's Neural Engine using reverse-engineered pri…
447275experimental
Marian NMT
Marian is an efficient Neural Machine Translation framework written in pure C++ with minimal dependencies, supporting fast multi-GPU traini…
231474maintenance
nerfstudio
NerfAcc is a PyTorch acceleration toolbox for Neural Radiance Fields (NeRF), focused on efficient sampling in the volumetric rendering pipe…
231456maintenance
TSDF Fusion
A lightweight Python script that fuses multiple registered RGB-D images into a projective TSDF voxel volume, from which high-quality 3D sur…
321432maintenance
vlfeat/matconvnet
MatConvNet is a MATLAB toolbox implementing convolutional neural networks (CNNs) for computer vision applications. It supports training and…
321430maintenance
dbolya/tomesd
ToMe for SD is a Python/PyTorch library that speeds up Stable Diffusion image generation by merging redundant tokens in the underlying tran…
211400maintenance
diwi/PixelFlow
PixelFlow is a Processing/Java library for high-performance GPU computing via GLSL shaders. It provides fluid simulation, flow-field partic…
231284maintenance
HackerPoet/FractalSoundExplorer
An interactive audio-visual sandbox application for exploring fractals in real time on the GPU. Users can click on fractals like the Mandel…
321250maintenance
enazoe/yolo-tensorrt
A C++ wrapper around NVIDIA TensorRT for running YOLO object detection models (YOLOv3, YOLOv4, YOLOv5) with support for FP32, FP16, and INT…
541201maintenance
hotshotco/Hotshot-XL
Hotshot-XL is an AI text-to-GIF model built to work alongside Stable Diffusion XL, generating 1-second GIFs at 8 FPS. It supports any fine-…
281116maintenance
mapbox/webgl-wind
A WebGL-powered library for visualizing wind data as GPU-accelerated particle animations, capable of rendering up to 1 million particles at…
721112maintenance
keroro824/HashingDeepLearning
SLIDE is a C++ research codebase implementing locality-sensitive hashing based training of deep neural networks, from the paper 'In Defense…
321104maintenance
Kismuz/btgym
BTGym is a Python library that wraps the Backtrader algorithmic trading engine in an OpenAI Gym environment API, enabling event-driven back…
321037maintenance
microsoft/nnfusion
NNFusion is a flexible and efficient deep neural network (DNN) compiler that generates high-performance executables from model descriptions…
231005maintenance
0hq/WebGPT
WebGPT is a vanilla JavaScript and HTML implementation of GPT transformer inference running in the browser via WebGPU, in under ~1500 lines…
303787experimental
evilsocket/cake
Cake is a multimodal AI inference server written in Rust that runs text, image, and voice models on a single device or shards them across a…
563126experimental
google-research/dex-lang
Dex is an experimental research programming language from Google Research for typed, functional array processing in the Haskell/ML family. …
561704experimental
collabora/WhisperFusion
WhisperFusion is a real-time voice chat application that combines WhisperLive speech-to-text, a Mistral/Phi LLM, and WhisperSpeech text-to-…
271648experimental
pfnet/PaintsChainer
Paints Chainer is a semi-automatic line drawing colorizer that uses convolutional neural networks built on the Chainer deep learning framew…
323760abandoned
mitsuba-renderer/mitsuba2
Mitsuba 2 is a research-oriented, retargetable physically-based rendering system written in C++17, supporting CPU, SIMD-vectorized, and GPU…
232073abandoned
intel/intel-extension-for-pytorch
A Python package extending official PyTorch with Intel-specific optimizations for CPUs and GPUs, including quantization and LLM inference a…
102012abandoned
kuz/DeepMind-Atari-Deep-Q-Learner
The original DQN 3.0 source code published by DeepMind alongside the 2015 Nature paper 'Human-level control through deep reinforcement lear…
321830abandoned
soumith/dcgan.torch
A Torch (Lua) implementation of the DCGAN paper for training deep convolutional generative adversarial networks that generate images. It su…
321488abandoned
nod-ai/AMD-SHARK-Studio
AMD-SHARK Studio is a web UI distribution for high-performance machine learning inference built on SHARK and IREE, primarily for running St…
451451abandoned
NiftyNet
NiftyNet is a TensorFlow-based open-source convolutional neural networks platform for research in medical image analysis and image-guided t…
101374abandoned
mitsuba-renderer/enoki
Enoki is a C++17 header-only template library for structured vectorization and automatic differentiation of numerical code, compiling the s…
101290abandoned
antonmks/Alenka
Alenka is a GPU-based column-store database engine written in CUDA that leverages vectorized processing and GPU memory bandwidth for fast a…
321166abandoned

← prev page 6 / 6