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 |
|---|---|---|---|
| ShaoqingRen/faster_rcnn A MATLAB re-implementation of Faster R-CNN, a deep-learning object detection framework combining a Region Proposal Network with a detection… | 32 | 2835 | abandoned |
| zzw922cn/Automatic_Speech_Recognition An end-to-end automatic speech recognition system implemented in TensorFlow, supporting Mandarin and English with models like DeepSpeech2, … | 32 | 2833 | abandoned |
| Neural Network Libraries (NNabla) Sony's Neural Network Libraries (nnabla) is a deep learning framework with a Python API over a C++11 core, designed for research, developme… | 10 | 2773 | abandoned |
| lisa-lab/pylearn2 Pylearn2 is a Python machine learning research library built on top of Theano, developed by the LISA lab at Université de Montréal. It prov… | 32 | 2765 | abandoned |
| kaonashi-tyc/zi2zi zi2zi is a TensorFlow implementation of a conditional GAN (based on pix2pix) for learning and transferring Chinese and East Asian character… | 32 | 2752 | abandoned |
| allanzelener/YAD2K YAD2K is a 90% Keras / 10% TensorFlow implementation and converter for YOLO_v2 object detection models. It converts Darknet cfg and weight … | 32 | 2728 | abandoned |
| NVlabs/MUNIT Official PyTorch implementation of MUNIT (ECCV 2018), a GAN-based model for multimodal unsupervised image-to-image translation. The reposit… | 23 | 2706 | abandoned |
| intel/BigDL BigDL is Intel's distributed deep learning library that scales TensorFlow, Keras, and PyTorch workloads on Apache Spark, Flink, and Ray, wi… | 10 | 2699 | abandoned |
| propelml/propel Propel was a JavaScript library for differentiable programming, providing tensor operations and automatic differentiation for scientific co… | 10 | 2690 | abandoned |
| ryanjay0/miles-deep Miles Deep is a C++ deep learning application built on Caffe that classifies each second of a pornographic video into six sexual act catego… | 23 | 2662 | abandoned |
| sherjilozair/char-rnn-tensorflow A Python/TensorFlow implementation of multi-layer recurrent neural networks (LSTM) for character-level language modeling, inspired by Andre… | 32 | 2658 | abandoned |
| coqui-ai/STT Coqui STT is an open-source deep learning toolkit for training and deploying speech-to-text models, built on TensorFlow with bindings for m… | 23 | 2605 | abandoned |
| devsisters/DQN-tensorflow A TensorFlow implementation of the Deep Q-Network (DQN) algorithm from the DeepMind paper 'Human-Level Control through Deep Reinforcement L… | 32 | 2586 | abandoned |
| rbgirshick/rcnn The original R-CNN (Region-based Convolutional Neural Networks) object detection system from UC Berkeley, released as research code accompa… | 23 | 2416 | abandoned |
| tryolabs/luminoth Luminoth is an open-source Python toolkit for computer vision built on TensorFlow and Sonnet, focused on object detection with Faster R-CNN… | 23 | 2406 | abandoned |
| kaishengtai/neuralart A Torch7 (Lua) implementation of the paper 'A Neural Algorithm of Artistic Style' for neural style transfer, applying the style of one imag… | 32 | 2394 | abandoned |
| facebookarchive/fb.resnet.torch A Torch (Lua) implementation of ResNet residual networks for image classification, with training scripts for ImageNet and pretrained models… | 10 | 2361 | abandoned |
| microsoft/Deep3DFaceReconstruction A TensorFlow implementation of a weakly-supervised CNN method for reconstructing accurate 3D face models (shape, texture, pose, landmarks) … | 32 | 2357 | abandoned |
| AutoAWQ AutoAWQ is a Python library implementing the Activation-aware Weight Quantization (AWQ) algorithm for 4-bit quantization of large language … | 10 | 2346 | abandoned |
| achillesrasquinha/bulbea A Python library for stock market prediction and modelling using deep learning, built on TensorFlow with RNN models for price forecasting a… | 32 | 2334 | abandoned |
| openai/improved-gan Reference implementation of the techniques from the OpenAI paper 'Improved Techniques for Training GANs' (arXiv:1606.03498), including semi… | 10 | 2332 | abandoned |
| isl-org/DPT DPT (Dense Prediction Transformers) is Intel's implementation of Vision Transformers for dense prediction tasks, providing pretrained model… | 10 | 2330 | abandoned |
| openai/finetune-transformer-lm Reference code and model for OpenAI's paper 'Improving Language Understanding by Generative Pre-Training' (the original GPT-1), implemented… | 10 | 2307 | abandoned |
| gluon-api/gluon-api Gluon API is a clear, concise, and efficient interface for building deep learning models with an imperative programming style, originally d… | 10 | 2293 | abandoned |
| facebookresearch/InferSent InferSent is a sentence embedding method from Facebook AI Research that produces semantic representations for English sentences using a mod… | 10 | 2281 | abandoned |
| PetrochukM/PyTorch-NLP PyTorch-NLP (torchnlp) is a Python library of basic utilities for natural language processing built on PyTorch, providing text data loading… | 10 | 2220 | abandoned |
| Pointer-Generator Networks (Get To The Point) Reference implementation of the pointer-generator network from the ACL 2017 paper 'Get To The Point: Summarization with Pointer-Generator N… | 32 | 2191 | abandoned |
| paarthneekhara/text-to-image An experimental TensorFlow implementation of text-to-image synthesis using Skip Thought Vectors and the GAN-CLS algorithm from the Generati… | 32 | 2164 | abandoned |
| facebookresearch/SparseConvNet A PyTorch library for training Submanifold Sparse Convolutional Networks, providing spatially-sparse convolutions that operate efficiently … | 10 | 2145 | abandoned |
| neuralmagic/sparseml SparseML is a Python library for applying sparsification recipes (pruning, quantization, sparsity) to neural networks with a few lines of c… | 10 | 2144 | abandoned |
| facebookresearch/darkforestGo DarkForest is a computer Go game engine developed at Facebook AI Research, powered by deep neural networks and Monte Carlo tree search. It … | 10 | 2096 | abandoned |
| openai/image-gpt OpenAI's official code and pre-trained models for iGPT (image GPT), a GPT-2-style transformer adapted to generate and classify images as pi… | 10 | 2094 | abandoned |
| LantaoYu/SeqGAN A Python/TensorFlow implementation of SeqGAN, the Sequence Generative Adversarial Nets with Policy Gradient model from the AAAI-17 paper. I… | 32 | 2092 | abandoned |
| ajbrock/Neural-Photo-Editor A GUI application for editing natural photos using generative neural networks (GANs/VAEs), implementing the Introspective Adversarial Netwo… | 32 | 2076 | abandoned |
| ryankiros/skip-thoughts A Python implementation of the Skip-Thought Vectors paper, providing a sent2vec encoder that maps sentences to fixed 4800-dimensional vecto… | 32 | 2048 | abandoned |
| 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… | 10 | 2012 | abandoned |
| facebookresearch/video-nonlocal-net A research codebase from Facebook AI implementing Non-local Neural Networks for video classification, built on the Caffe2 framework. It rep… | 10 | 1990 | abandoned |
| keras-team/keras-applications Keras Applications provides reference implementations of popular deep learning models (e.g., ResNet, VGG, MobileNet) for use with Keras. Th… | 10 | 1989 | abandoned |
| openai/pixel-cnn A Python/TensorFlow implementation of PixelCNN++, an autoregressive generative model for images with discretized logistic mixture likelihoo… | 10 | 1957 | abandoned |
| Po-Hsun-Su/pytorch-ssim A small PyTorch library providing a differentiable structural similarity (SSIM) index and SSIM loss module for image comparison. It can be … | 32 | 1949 | abandoned |
| facebookresearch/ResNeXt A Torch (Lua) implementation of the ResNeXt architecture from the paper 'Aggregated Residual Transformations for Deep Neural Networks', bui… | 10 | 1927 | abandoned |
| U-Net Tensorflow implementation A generic U-Net implementation built on TensorFlow 1.x for training image segmentation models on arbitrary imaging data. Originally develop… | 23 | 1909 | abandoned |
| jiesutd/LatticeLSTM A PyTorch implementation of Lattice LSTM for Chinese Named Entity Recognition, accompanying an ACL 2018 paper. It combines character-based … | 32 | 1832 | abandoned |
| 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… | 32 | 1830 | abandoned |
| tensorflow/fold TensorFlow Fold is a Python library for building TensorFlow models that consume structured data, where the computation graph structure depe… | 10 | 1816 | abandoned |
| ONNX.js ONNX.js is a JavaScript library for running pre-trained ONNX machine learning models directly in browsers and Node.js, using WebAssembly fo… | 10 | 1815 | abandoned |
| ruotianluo/pytorch-faster-rcnn A PyTorch 1.0 implementation of the Faster R-CNN object detection framework, based on Xinlei Chen's tf-faster-rcnn, supporting VGG16, ResNe… | 23 | 1810 | abandoned |
| NVlabs/few-shot-vid2vid A PyTorch implementation of few-shot photorealistic video-to-video translation from NVIDIA Research (NeurIPS 2019), generating realistic vi… | 32 | 1795 | abandoned |
| keplr-io/quiver Quiver is a Python library (quiver_engine) that launches an interactive web dashboard to visualize layer activations of Keras convolutional… | 32 | 1784 | abandoned |
| dunnousername/yanderifier A GUI front-end wrapper around the first-order-motion model that animates a still face image to match the motion of a driving video. It bun… | 23 | 1781 | abandoned |
| longcw/faster_rcnn_pytorch A PyTorch re-implementation of Faster R-CNN for object detection, based on the original py-faster-rcnn and TFFRCNN projects. It supports tr… | 32 | 1779 | abandoned |
| traveller59/second.pytorch A PyTorch implementation of the SECOND 3D object detector for LiDAR point clouds, supporting KITTI and NuScenes datasets with sparse convol… | 32 | 1777 | abandoned |
| tomlepaine/fast-wavenet A Python/TensorFlow library implementing an efficient O(L) generation algorithm for Wavenet-style autoregressive models using dynamic progr… | 32 | 1769 | abandoned |
| gram-ai/capsule-networks A barebones CUDA-enabled PyTorch implementation of the CapsNet architecture from the NIPS 2017 paper 'Dynamic Routing Between Capsules' by … | 32 | 1757 | abandoned |
| TuSimple/mx-maskrcnn An MXNet implementation of the Mask R-CNN instance segmentation model, built on top of the mx-rcnn Faster R-CNN codebase. It includes train… | 32 | 1753 | abandoned |
| facebookarchive/MemNN Facebook AI Research's reference implementations of memory-augmented neural networks (MemN2N, Key-Value Memory Networks, Entity Networks) w… | 10 | 1753 | abandoned |
| dog-qiuqiu/MobileNet-Yolo A collection of ultra-lightweight YOLOv3-based object detection models (MobileNetV2-YOLOv3-Lite/Nano, YoloFace) designed for mobile and emb… | 32 | 1745 | abandoned |
| lukalabs/cakechat CakeChat is a Python backend for building emotional generative chatbots using a hierarchical recurrent encoder-decoder (seq2seq) architectu… | 10 | 1708 | abandoned |
| huggingface/swift-coreml-transformers A Swift library with Core ML implementations of GPT-2, DistilGPT-2, BERT, and DistilBERT, including tokenizers and demo apps for on-device … | 10 | 1683 | abandoned |
| nyu-mll/jiant jiant is a PyTorch-based NLP research toolkit for multitask and transfer learning, supporting 50+ natural language understanding tasks and … | 23 | 1674 | abandoned |
| ramprs/grad-cam Official Torch (Lua) implementation of Grad-CAM, the ICCV 2017 gradient-weighted class activation mapping technique for producing visual ex… | 32 | 1671 | abandoned |
| terrifyzhao/bert-utils A Python library that simplifies Google's BERT to generate sentence vectors with one line of code and perform text classification and text … | 32 | 1669 | abandoned |
| ry/tensorflow-resnet An implementation of the ResNet deep residual learning model in TensorFlow, including a tool to convert Kaiming He's published Caffe pretra… | 10 | 1665 | abandoned |
| NNVM NNVM was a computation-graph intermediate representation and compiler front-end for deep learning frameworks, serving as the graph-level op… | 10 | 1649 | abandoned |
| openai/sparse_attention A Python library of sparse attention primitives from OpenAI's Sparse Transformers paper, including fused normal attention and strided/fixed… | 10 | 1612 | abandoned |
| facebookresearch/ClassyVision Classy Vision is an end-to-end PyTorch framework for image and video classification, offering a modular API for computer vision research wi… | 10 | 1607 | abandoned |
| jcjohnson/densecap DenseCap is a Torch implementation of a fully convolutional localization network that performs dense captioning: detecting objects in image… | 32 | 1599 | abandoned |
| keras-team/keras-contrib Keras-contrib is the official community extension repository for the Keras deep learning library, providing additional layers, activations,… | 10 | 1586 | abandoned |
| ELS-RD/kernl Kernl is an open-source PyTorch model optimization library that accelerates transformer inference on GPUs with a single line of code, using… | 10 | 1583 | abandoned |
| carpedm20/deep-rl-tensorflow A TensorFlow implementation of several Deep Reinforcement Learning papers, including DQN, Double DQN, and Dueling networks, for training ag… | 32 | 1580 | abandoned |
| rwightman/gen-efficientnet-pytorch A PyTorch library providing generic implementations and pretrained weights for compute-efficient image classification models such as Effici… | 32 | 1576 | abandoned |
| xunhuang1995/AdaIN-style Torch (Lua) implementation of the ICCV 2017 paper 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization'. It applies … | 32 | 1569 | abandoned |
| longcw/yolo2-pytorch A PyTorch implementation of the YOLOv2 real-time object detection model, based on darknet and darkflow. It includes training and inference … | 32 | 1561 | abandoned |
| NVIDIA/OpenSeq2Seq OpenSeq2Seq is a TensorFlow-based toolkit for building and training sequence-to-sequence models for neural machine translation, speech reco… | 10 | 1559 | abandoned |
| wizyoung/YOLOv3_TensorFlow A pure TensorFlow implementation of the YOLOv3 real-time object detection model, including weight conversion from darknet, training, and ev… | 23 | 1550 | abandoned |
| fzliu/style-transfer A Python implementation of the Gatys et al. neural style transfer algorithm built on Caffe, applying the artistic style of one image onto a… | 32 | 1534 | abandoned |
| huggingface/pytorch-openai-transformer-lm A PyTorch reimplementation of OpenAI's finetuned transformer language model (GPT-1) from the paper 'Improving Language Understanding by Gen… | 10 | 1525 | abandoned |
| openai/generating-reviews-discovering-sentiment Research code accompanying the OpenAI paper 'Learning to Generate Reviews and Discovering Sentiment', providing a byte-level multiplicative… | 10 | 1520 | abandoned |
| chenriwei/DeepFace A deep learning face analysis toolkit built on Caffe, implementing face detection, facial keypoint (alignment) detection, and face verifica… | 32 | 1507 | abandoned |
| soumith/dcgan.torch A Torch (Lua) implementation of the DCGAN paper for training deep convolutional generative adversarial networks that generate images. It su… | 32 | 1488 | abandoned |
| 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… | 45 | 1451 | abandoned |
| dhwajraj/deep-siamese-text-similarity A TensorFlow implementation of a deep siamese LSTM network that learns phrase and sentence similarity using character-level and word-level … | 32 | 1414 | abandoned |
| raghakot/keras-resnet A Keras implementation of Residual Networks (ResNet) using the Keras 1.0 functional API, supporting both Theano and TensorFlow backends. It… | 32 | 1389 | abandoned |
| bonlime/keras-deeplab-v3-plus A Keras implementation of the DeepLab v3+ semantic image segmentation model with pretrained weights imported from the original TensorFlow c… | 23 | 1374 | abandoned |
| NiftyNet NiftyNet is a TensorFlow-based open-source convolutional neural networks platform for research in medical image analysis and image-guided t… | 10 | 1374 | abandoned |
| TorchSSL/TorchSSL TorchSSL is a PyTorch-based toolbox for semi-supervised learning that implements popular SSL algorithms like FixMatch, FlexMatch, MixMatch,… | 32 | 1371 | abandoned |
| andersbll/deeppy DeepPy is a Pythonic deep learning framework built on top of NumPy, with optional CUDA acceleration via the cudarray library. It provides a… | 32 | 1371 | abandoned |
| MIC-DKFZ/medicaldetectiontoolkit A PyTorch framework providing 2D and 3D implementations of object detectors like Mask R-CNN, Retina Net, and Retina U-Net, tailored for med… | 32 | 1361 | abandoned |
| reiinakano/fast-style-transfer-deeplearnjs A browser-based demo of Fast Neural Style Transfer running entirely client-side with the deeplearn.js library, letting users stylize images… | 32 | 1350 | abandoned |
| ericjang/tdb TensorDebugger (TDB) is a visual debugger for TensorFlow deep learning models, combining a Python library with a Jupyter notebook extension… | 23 | 1349 | abandoned |
| NervanaSystems/ngraph nGraph is a C++ deep learning graph compiler that accelerates neural network workloads from frameworks like TensorFlow, ONNX, MXNet, and Py… | 10 | 1344 | abandoned |
| facebookresearch/multipathnet A Torch-7 implementation of the MultiPath Network for object detection from the BMVC 2016 paper by Facebook AI Research, also supporting Fa… | 10 | 1331 | abandoned |
| onnx/onnx-tensorflow A Python library and CLI that converts ONNX machine learning models into TensorFlow models so they can be executed with TensorFlow. It is t… | 64 | 1329 | abandoned |
| IDSIA/brainstorm Brainstorm is a Python neural network library designed to be fast, flexible, and fun, supporting multiple computing backends. The project h… | 32 | 1326 | abandoned |
| bamos/dcgan-completion.tensorflow A TensorFlow implementation of semantic image inpainting using DCGANs, based on the paper 'Semantic Image Inpainting with Perceptual and Co… | 32 | 1317 | abandoned |
| happynear/caffe-windows A Windows-specific fork of the Caffe deep learning framework with preconfigured Visual Studio project files, enabling users to build Caffe … | 23 | 1310 | abandoned |
| datitran/object_detector_app A Python application that performs real-time object recognition from a webcam or HLS video stream using TensorFlow's Object Detection API a… | 32 | 1306 | abandoned |
| piiswrong/deep3d Deep3D is a CNN-based research project that automatically converts 2D images and videos into 3D by estimating per-pixel depth maps and gene… | 32 | 1299 | abandoned |
| hunkim/word-rnn-tensorflow A TensorFlow implementation of multi-layer recurrent neural networks (LSTM/RNN) for training word-level language models, adapted from char-… | 32 | 1294 | abandoned |
| tianzhi0549/CTPN Reference implementation of CTPN (Connectionist Text Proposal Network) for detecting text lines in natural images, from the ECCV 2016 paper… | 32 | 1286 | abandoned |