Open-source deep-learning software
2785 open-source libraries, tools and applications tagged deep-learning, ranked by how closely deep-learning 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.
2785 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| chelsea0x3b/dfdx A Rust deep learning library providing shape-checked tensors and neural network building blocks with compile-time verification of tensor op… | 23 | 1934 | experimental |
| matthewearl/deep-anpr An experimental Python project that uses neural networks (TensorFlow) to build an automatic number plate recognition (ANPR) system. It incl… | 32 | 1860 | experimental |
| rentruewang/aioway Aioway is an optimizing compiler for deep learning algorithms that treats ML models as instructions and builds declarative pipelines using … | 77 | 1823 | experimental |
| deepseek-ai/DeepSeek-V3.2-Exp DeepSeek-V3.2-Exp is an experimental open-weight large language model release that introduces DeepSeek Sparse Attention for more efficient … | 37 | 1654 | experimental |
| Anything-of-anything/Anything-3D Anything-3D is a Python research project that combines Meta's Segment Anything model with a series of 3D models (3DFuse, Zero 1-to-3, NeRF,… | 30 | 1633 | experimental |
| EvolvingLMMs-Lab/open-r1-multimodal A fork of Hugging Face's open-r1 that extends the R1 GRPO reinforcement learning training paradigm to multimodal (vision-language) models l… | 21 | 1606 | experimental |
| lyuchenyang/Macaw-LLM Macaw-LLM is a multi-modal language modeling framework that integrates image, video, audio, and text data, built on CLIP, Whisper, and LLaM… | 30 | 1591 | experimental |
| lichao-sun/Mora Mora is an open-source multi-agent framework for generalist video generation, using multiple collaborative visual AI agents to replicate ca… | 26 | 1591 | experimental |
| AnswerDotAI/fsdp_qlora A training script/library from Answer.AI that combines QLoRA (quantized LoRA) with PyTorch FSDP to fine-tune large language models like Lla… | 27 | 1554 | experimental |
| KUR-creative/SickZil-Machine SickZil-Machine is a desktop application that automates text removal from manga and comic pages during the scanlation (translation) process… | 23 | 1526 | experimental |
| Learning to (Learn at Test Time) Language Model Official PyTorch implementation of Test-Time Training (TTT) layers, a sequence modeling layer with linear complexity whose hidden state is … | 24 | 1405 | experimental |
| lucidrains/video-diffusion-pytorch A Pytorch implementation of the Video Diffusion Models paper by Jonathan Ho, extending denoising diffusion probabilistic models (DDPMs) to … | 23 | 1391 | experimental |
| AutoArk/TinyEngram TinyEngram is an open research project exploring DeepSeek-AI's Engram architecture and memory injection as an alternative to LoRA for param… | 50 | 1380 | experimental |
| lizhihao6/Sparc3D Sparc3D is the official implementation of a research framework for high-resolution 3D shape modeling, combining a sparse deformable marchin… | 28 | 1359 | experimental |
| lucidrains/naturalspeech2-pytorch A PyTorch implementation of NaturalSpeech 2, a zero-shot text-to-speech and singing synthesizer that combines a neural audio codec with a l… | 21 | 1334 | experimental |
| google/style-aligned Official research code for 'Style Aligned Image Generation via Shared Attention', implementing style-consistent image generation with diffu… | 10 | 1315 | experimental |
| Jamie-Stirling/RetNet A minimal, pure PyTorch implementation of the Retentive Network (RetNet) architecture proposed as a successor to Transformers for large lan… | 29 | 1212 | experimental |
| tspeterkim/flash-attention-minimal A minimal ~100-line CUDA re-implementation of the Flash Attention forward pass, loadable as a PyTorch extension. It is designed as an educa… | 26 | 1192 | experimental |
| facebookresearch/shumai Shumai is a fast, differentiable tensor library for TypeScript and JavaScript built on Bun and Flashlight (ArrayFire backend). It provides … | 32 | 1171 | experimental |
| Ligo-Biosciences/AlphaFold3 An open-source Python implementation of AlphaFold3, DeepMind's biomolecular structure prediction model, including the full model architectu… | 27 | 1093 | experimental |
| thinking-machines-lab/batch_invariant_ops A Python library that replaces standard PyTorch CUDA kernels with batch-invariant versions, ensuring identical results regardless of batch … | 37 | 1080 | experimental |
| volcengine/veScale veScale is a PyTorch distributed training library from ByteDance for hyperscale training of large language models and reinforcement learnin… | 54 | 1041 | experimental |
| orca-wm/Orca Orca is a general world foundation model from BAAI centered on Next-State-Prediction, learning a unified world latent space from visual and… | 54 | 1021 | experimental |
| meta-llama/llama Meta's official repository containing inference code and download scripts for Llama 2 large language models (7B to 70B parameters). The rep… | 31 | 59632 | abandoned |
| facebookresearch/Detectron Facebook AI Research's Python software system implementing state-of-the-art object detection algorithms such as Mask R-CNN, RetinaNet, and … | 10 | 26352 | abandoned |
| openai/gpt-2 Official OpenAI repository containing code and pretrained model weights for GPT-2, from the paper 'Language Models are Unsupervised Multita… | 10 | 25009 | abandoned |
| apache/mxnet Apache MXNet is a deep learning framework offering a hybrid front-end that mixes imperative (Gluon) and symbolic programming, with a dynami… | 10 | 20804 | abandoned |
| Magenta Magenta is a Google Brain research project and Python TensorFlow library for generating music, images, and drawings with machine learning. … | 10 | 19787 | abandoned |
| jantic/DeOldify DeOldify is a deep learning project for colorizing and restoring old black-and-white images and videos, with artistic and stable model vari… | 10 | 18463 | abandoned |
| jcjohnson/neural-style A Torch (Lua) implementation of the Gatys et al. neural style transfer algorithm, which combines the content of one image with the artistic… | 32 | 18269 | abandoned |
| microsoft/CNTK The Microsoft Cognitive Toolkit (CNTK) is an open-source deep-learning toolkit that describes neural networks as directed computational gra… | 10 | 17584 | abandoned |
| horovod/horovod Horovod is a distributed deep learning training framework for TensorFlow, Keras, PyTorch, and Apache MXNet, originally developed at Uber. I… | 10 | 14680 | abandoned |
| Jiayi-Pan/TinyZero TinyZero is a minimal reproduction of DeepSeek R1-Zero, showing that a 3B base language model can develop self-verification and search abil… | 49 | 13248 | abandoned |
| alexjc/neural-enhance Neural Enhance is a Python command-line tool that upscales images 2x or 4x using deep learning super-resolution models. It ships with pre-t… | 10 | 11865 | abandoned |
| karpathy/convnetjs ConvNetJS is a JavaScript library for defining and training neural networks, including convolutional networks and an experimental Deep Q Le… | 23 | 11208 | abandoned |
| Theano/Theano Theano is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays… | 32 | 10002 | abandoned |
| facebookresearch/maskrcnn-benchmark A fast, modular PyTorch reference implementation of instance segmentation and object detection algorithms including Mask R-CNN, Faster R-CN… | 10 | 9361 | abandoned |
| voicepaw/so-vits-svc-fork A fork of so-vits-svc providing singing voice conversion with realtime support and an improved interface, built on PyTorch and PyTorch Ligh… | 81 | 9333 | abandoned |
| Torch7 Torch7 is a scientific computing framework with a fast multi-dimensional tensor library and math operations, originally built for Lua with … | 35 | 9143 | abandoned |
| intel/ipex-llm IPEX-LLM is a PyTorch LLM acceleration library for Intel hardware (iGPU, NPU, Arc/Flex/Max GPUs, and CPU), offering low-bit quantization (F… | 10 | 8845 | abandoned |
| TensorFlow.js tfjs-core was the core WebGL-accelerated tensor computation library behind TensorFlow.js, providing linear algebra operations and automatic… | 10 | 8425 | abandoned |
| facebookarchive/caffe2 Caffe2 is a lightweight, modular, and scalable deep learning framework built on the original Caffe, with both Python and C++ APIs. It has b… | 10 | 8370 | abandoned |
| OptiMate OptiMate is a collection of Python libraries from Nebuly AI for optimizing AI model performance, including Speedster for inference accelera… | 23 | 8328 | abandoned |
| rbgirshick/py-faster-rcnn A Python reimplementation of the Faster R-CNN object detection model built on a fork of Fast R-CNN and Caffe. The repository is officially … | 32 | 8289 | abandoned |
| EleutherAI/gpt-neo GPT-Neo is EleutherAI's implementation of model- and data-parallel GPT-3-style transformer language models built on mesh-tensorflow, with r… | 10 | 8255 | abandoned |
| openai/jukebox OpenAI's research code and pretrained model weights for Jukebox, a generative neural network that produces music with vocals as raw audio f… | 10 | 8028 | abandoned |
| jwyang/faster-rcnn.pytorch A pure PyTorch implementation of Faster R-CNN for object detection, supporting multi-image batch training and multi-GPU training with sever… | 32 | 7859 | abandoned |
| fchollet/deep-learning-models A deprecated collection of Keras code and pre-trained weights for popular deep learning image classification models such as VGG16, VGG19, R… | 23 | 7351 | abandoned |
| carpedm20/DCGAN-tensorflow A TensorFlow implementation of Deep Convolutional Generative Adversarial Networks (DCGAN) for generating images such as faces and digits. I… | 32 | 7178 | abandoned |
| facebookresearch/pytext PyText is a deep-learning-based NLP modeling framework built on PyTorch, providing extensible abstractions for text classifiers, sequence t… | 10 | 6291 | abandoned |
| Swift for TensorFlow Swift for TensorFlow was an experimental next-generation machine learning platform integrating differentiable programming directly into the… | 10 | 6121 | abandoned |
| Lightning-AI/lit-llama Lit-LLaMA is an independent, Apache 2.0-licensed implementation of the LLaMA language model built on nanoGPT, covering pre-training, fine-t… | 40 | 6080 | abandoned |
| Swift AI Swift AI is a high-performance deep learning library written entirely in Swift, offering a flexible fully-connected neural network optimize… | 23 | 6036 | abandoned |
| tiny-dnn/tiny-dnn tiny-dnn is a header-only, dependency-free deep learning framework written in C++14, designed for training and running neural networks on r… | 23 | 6028 | abandoned |
| salesforce/BLIP PyTorch implementation of BLIP, a vision-language pre-training model for unified image understanding and text generation tasks. The reposit… | 10 | 5708 | abandoned |
| google/seq2seq A general-purpose encoder-decoder (seq2seq) framework built on TensorFlow, supporting tasks like machine translation, text summarization, c… | 10 | 5619 | abandoned |
| karpathy/neuraltalk2 NeuralTalk2 is a Torch (Lua) implementation of image captioning that uses a CNN (VGGNet) followed by an RNN language model to generate text… | 32 | 5600 | abandoned |
| Leaf Leaf is an open-source machine learning framework written in Rust for building classical, deep, or hybrid ML applications, inspired by Tens… | 32 | 5540 | abandoned |
| karpathy/neuraltalk NeuralTalk is a Python+numpy implementation of Multimodal Recurrent Neural Networks that generate natural-language descriptions of images. … | 32 | 5508 | abandoned |
| ashnkumar/sketch-code A Keras/TensorFlow deep learning model that converts hand-drawn website wireframes into HTML code using an image captioning architecture (C… | 72 | 5141 | abandoned |
| Cube Studio Cube Studio is an open-source, cloud-native, one-stop machine learning platform covering the full MLOps lifecycle: online notebooks, drag-a… | 79 | 5079 | abandoned |
| AutoGPTQ/AutoGPTQ AutoGPTQ is a Python library for quantizing large language models using the GPTQ weight-only quantization algorithm, with user-friendly API… | 10 | 5065 | abandoned |
| transcranial/keras-js A JavaScript library for running trained Keras deep learning models directly in the browser, with GPU acceleration via WebGL 2. It also sup… | 23 | 4964 | abandoned |
| DenseNet Reference implementation of DenseNet (Densely Connected Convolutional Networks), the CVPR 2017 Best Paper Award-winning CNN architecture, w… | 32 | 4872 | abandoned |
| huggingface/autotrain-advanced AutoTrain Advanced is Hugging Face's no-code tool for training and deploying state-of-the-art machine learning models, covering LLM finetun… | 76 | 4612 | abandoned |
| microsoft/VoTT VoTT (Visual Object Tagging Tool) is an open-source Electron application for annotating and labeling images and video frames to build objec… | 10 | 4427 | abandoned |
| amazon-archives/amazon-dsstne DSSTNE (Deep Scalable Sparse Tensor Network Engine) is a C++ GPU library from Amazon for training and deploying deep learning recommendatio… | 10 | 4390 | abandoned |
| RetinaNet (Keras/TF implementation) A Keras/TensorFlow implementation of the RetinaNet object detection model with focal loss, supporting training and inference on custom data… | 23 | 4382 | abandoned |
| janhuenermann/neurojs neurojs is a JavaScript deep learning framework for the browser with a focus on reinforcement learning, supporting deep Q-networks and acto… | 32 | 4367 | abandoned |
| NVIDIA/DIGITS DIGITS is a web application for training deep learning models on GPUs, supporting frameworks like Caffe, Torch, and TensorFlow with a brows… | 10 | 4176 | abandoned |
| goodfeli/adversarial Reference implementation of the original 2014 'Generative Adversarial Networks' paper by Goodfellow et al., containing training code, YAML … | 32 | 4087 | abandoned |
| google-deepmind/learning-to-learn A TensorFlow implementation of 'Learning to learn by gradient descent by gradient descent' (DeepMind, arXiv:1606.04474), which trains LSTM-… | 32 | 4072 | abandoned |
| junyanz/iGAN iGAN is a research application implementing interactive image generation with generative adversarial networks, letting users draw strokes t… | 32 | 4012 | abandoned |
| buriburisuri/speech-to-text-wavenet A TensorFlow implementation of end-to-end English speech recognition based on DeepMind's WaveNet architecture, trained with CTC loss on sen… | 32 | 4003 | abandoned |
| riffusion/riffusion-hobby Riffusion is a Python library for real-time music and audio generation using stable diffusion, converting between spectrogram images and au… | 23 | 3900 | abandoned |
| rasmusbergpalm/DeepLearnToolbox A Matlab/Octave toolbox implementing classic deep learning models including Deep Belief Nets, Stacked Autoencoders, Convolutional Neural Ne… | 32 | 3869 | abandoned |
| NervanaSystems/neon neon was Intel Nervana's reference deep learning framework in Python, focused on best-in-class training performance on CPUs, GPUs, and Nerv… | 10 | 3860 | abandoned |
| facebookresearch/fairseq-lua Fairseq-lua is Facebook AI Research's sequence-to-sequence learning toolkit for the Torch framework, focused on neural machine translation … | 10 | 3725 | abandoned |
| fo40225/tensorflow-windows-wheel A repository of prebuilt TensorFlow wheel binaries for 64-bit Windows, including CPU-only and CUDA/cuDNN GPU builds with AVX2 and SSE2 (non… | 32 | 3665 | abandoned |
| endernewton/tf-faster-rcnn A TensorFlow implementation of the Faster R-CNN object detection framework, supporting VGG16, ResNet, and MobileNet backbones trained and e… | 23 | 3649 | abandoned |
| apple/tensorflow_macos A pre-release fork of TensorFlow and TensorFlow Addons for macOS 11.0+ that provides hardware acceleration on Apple Silicon and Intel Macs … | 10 | 3643 | abandoned |
| pytorch/text TorchText is a PyTorch companion library providing datasets, data loaders, vocabularies, text transforms, and pre-trained models for natura… | 10 | 3553 | abandoned |
| pavelgonchar/colornet Colornet is a Python/TensorFlow neural network implementation that colorizes grayscale images, based on VGG16 features and YUV color-channe… | 32 | 3550 | abandoned |
| Newmu/dcgan_code Reference implementation of the DCGAN paper (Unsupervised Representation Learning with Deep Convolutional GANs) in Python. It trains convol… | 32 | 3529 | abandoned |
| rbgirshick/fast-rcnn Fast R-CNN is Ross Girshick's ICCV 2015 framework for fast object detection with deep convolutional networks, written in Python and C++/Caf… | 32 | 3462 | abandoned |
| tensorforce/tensorforce Tensorforce is an open-source deep reinforcement learning framework built on TensorFlow, emphasizing modular, configurable library design f… | 64 | 3305 | abandoned |
| latitudegames/AIDungeon AI Dungeon 2 is an open-source, infinitely generated text adventure game powered by a fine-tuned GPT-2 language model. It can be played loc… | 10 | 3220 | abandoned |
| openai/glow Official research code from OpenAI for reproducing the 'Glow: Generative Flow with Invertible 1x1 Convolutions' paper, a flow-based generat… | 10 | 3181 | abandoned |
| alpa-projects/alpa Alpa is a Python system for training and serving large-scale neural networks by automatically parallelizing single-device code across distr… | 10 | 3176 | abandoned |
| tensorflow/skflow A simplified Scikit Learn-style interface for TensorFlow deep learning, providing estimators for neural networks. It was merged into Tensor… | 10 | 3165 | abandoned |
| neuralmagic/deepsparse DeepSparse is a sparsity-aware deep learning inference runtime that delivers GPU-class performance on x86 CPUs for ONNX models, covering co… | 10 | 3157 | abandoned |
| LeeJunHyun/Image_Segmentation A PyTorch implementation of four U-Net variants for image segmentation: U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net. It includes… | 32 | 3102 | abandoned |
| facebookresearch/deepmask A Torch (Lua) implementation of the DeepMask and SharpMask object proposal algorithms from Facebook AI Research. It generates class-agnosti… | 10 | 3097 | abandoned |
| mistralai/mistral-finetune A lightweight Python codebase from Mistral AI for memory-efficient LoRA fine-tuning of Mistral's language models, optimized for single-node… | 10 | 3094 | abandoned |
| CharlesShang/FastMaskRCNN A TensorFlow implementation of Mask R-CNN for instance segmentation, reproducing the paper by Kaiming He et al. It includes ROIAlign, a COC… | 32 | 3082 | abandoned |
| raghakot/keras-vis keras-vis is a high-level Python toolkit for visualizing and debugging trained Keras neural network models. It supports activation maximiza… | 23 | 2992 | abandoned |
| ryankiros/neural-storyteller A recurrent neural network research project that generates short stories about images by combining skip-thought vectors, visual-semantic em… | 32 | 2959 | abandoned |
| IntelLabs/nlp-architect NLP Architect is an open-source Python library from Intel AI Lab for exploring state-of-the-art deep learning topologies and optimization t… | 10 | 2931 | abandoned |
| jisungk/deepjazz deepjazz is a Python application that uses a two-layer LSTM built with Keras and Theano to generate jazz music from a given MIDI file. It w… | 61 | 2895 | abandoned |
| isl-org/ZoeDepth ZoeDepth is a PyTorch library implementing metric depth estimation from a single image, combining relative and metric depth approaches with… | 10 | 2838 | abandoned |