Project Awesome project awesome

Deep Learning

Neural networks.

Collection 27.8k stars GitHub

Researchers

Frameworks

convetjs 11.1k updated 3y ago
Brain 8.0k (archived)
DeepLearnToolbox 3.9k updated 7y ago
Deepnet 896 updated 11y ago
Deeppy 1.4k updated 5y ago
JavaNN 1.2k updated 8y ago
hebel
Mocha.jl 1.3k updated 7y ago
OpenDL 222 updated 10mo ago
Knet.jl 1.4k updated 1y ago
Nvidia DIGITS - a web app based on Caffe 4.2k (archived)
Neon - Python based Deep Learning Framework 3.9k (archived)
char-rnn 12.0k updated 2y ago
MatConvNet: CNNs for MATLAB 1.4k updated 4y ago
Minerva - a fast and flexible tool for deep learning on multi-GPU 707 (archived)
Brainstorm - Fast, flexible and fun neural networks. 1.3k updated 3y ago
Tensorflow - Open source software library for numerical computation using data flow graphs
DMTK - Microsoft Distributed Machine Learning Tookit 2.7k (archived)
Scikit Flow - Simplified interface for TensorFlow (mimicking Scikit Learn)
MXnet - Lightweight, Portable, Flexible Distributed/Mobile Deep Learning framework 20.8k (archived)
Veles - Samsung Distributed machine learning platform 916 updated 2y ago
Marvin - A Minimalist GPU-only N-Dimensional ConvNets Framework
DSSTNE - Amazon's library for building Deep Learning models 4.4k (archived)
Torchnet - Torch based Deep Learning Library 992 (archived)
Paddle - PArallel Distributed Deep LEarning by Baidu 23.8k updated 2d ago
Lasagne - a lightweight library to build and train neural networks in Theano 3.9k updated 4y ago
nolearn - wrappers and abstractions around existing neural network libraries, most notably Lasagne 955 updated 2y ago
Sonnet - a library for constructing neural networks by Google's DeepMind 9.9k updated 1mo ago
PyTorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration 98.5k updated 2d ago
CNTK - Microsoft Cognitive Toolkit 17.6k (archived)
Serpent.AI - Game agent framework: Use any video game as a deep learning sandbox
Caffe2 - A New Lightweight, Modular, and Scalable Deep Learning Framework
deeplearn.js - Hardware-accelerated deep learning and linear algebra (NumPy) library for the web 8.5k (archived)
Coach - Reinforcement Learning Coach by Intel AI Lab
albumentations - A fast and framework agnostic image augmentation library 15.3k (archived)
Neuraxle - A general-purpose ML pipelining framework 614 updated 1mo ago
Catalyst: High-level utils for PyTorch DL & RL research. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing 3.4k updated 9mo ago
garage - A toolkit for reproducible reinforcement learning research 2.1k updated 2y ago
Detecto - Train and run object detection models with 5-10 lines of code 625 updated 1y ago
Karate Club - An unsupervised machine learning library for graph structured data 2.3k updated 1y ago
Synapses - A lightweight library for neural networks that runs anywhere 73 updated 4y ago
TensorForce - A TensorFlow library for applied reinforcement learning 3.3k updated 1y ago
Hopsworks - A Feature Store for ML and Data-Intensive AI 1.3k updated 1y ago
PyTorch Geometric Temporal - Representation learning on dynamic graphs 6.8k updated 2d ago
lightly - A computer vision framework for self-supervised learning 3.7k updated 2d ago
Trax — Deep Learning with Clear Code and Speed 8.3k (archived)
Flax - a neural network ecosystem for JAX that is designed for flexibility
QuickVision 53 updated 2y ago
Colossal-AI - An Integrated Large-scale Model Training System with Efficient Parallelization Techniques 41.4k updated 9d ago
Maze 287 updated 5d ago

Application-oriented deep reinforcement learning framework addressing real-world decision problems.

InsNet - A neural network library for building instance-dependent NLP models with padding-free dynamic batching 67 updated 4y ago

Miscellaneous

Caffe DockerFile
TorontoDeepLearning convnet 505 updated 5y ago
gfx.js 125 updated 8y ago
Torch7 Cheat sheet 9.1k updated 11mo ago
A chess AI that learns to play chess using deep learning. 829 updated 9y ago
Reproducing the results of "Playing Atari with Deep Reinforcement Learning" by DeepMind 667 updated 11y ago
Wiki2Vec. Getting Word2vec vectors for entities and word from Wikipedia Dumps 601 (archived)
The original code from the DeepMind article + tweaks 1.8k updated 8y ago
Google deepdream - Neural Network art 13.2k (archived)
An efficient, batched LSTM.
A recurrent neural network designed to generate classical music. 1.9k updated 2y ago
Memory Networks Implementations - Facebook
Face recognition with Google's FaceNet deep neural network. 15.4k updated 1y ago
Basic digit recognition neural network
Proof of concept for loading Caffe models in TensorFlow 2.8k updated 6y ago
AlphaGo - A replication of DeepMind's 2016 Nature publication, "Mastering the game of Go with deep neural networks and tree search"
Machine Learning for Software Engineers
Dockerface 191 updated 5y ago

Easy to install and use deep learning Faster R-CNN face detection for images and video in a docker container.

Awesome Deep Learning Music 3.0k updated 2y ago

Curated list of articles related to deep learning scientific research applied to music

Awesome Graph Embedding 4.8k updated 3y ago

Curated list of articles related to deep learning scientific research on graph structured data at the graph level.

Awesome Network Embedding

Curated list of articles related to deep learning scientific research on graph structured data at the node level.

Microsoft Recommenders 21.5k updated 3d ago

contains examples, utilities and best practices for building recommendation systems. Implementations of several state-of-the-art algorithms are provided for self-study and customization in your own applications.

Ladder Network 99 updated 4y ago

Keras Implementation of Ladder Network for Semi-Supervised Learning

toolbox: Curated list of ML libraries
AI Expert Roadmap 30.9k updated 6mo ago

Roadmap to becoming an Artificial Intelligence Expert

Awesome Drug Interactions, Synergy, and Polypharmacy Prediction 98 updated 3y ago

Table of Contents

Magenta 19.8k (archived)

Magenta is a project by Google exploring the role of machine learning in the process of creating art and music.

caffe-net

The CaffeNet model, a popular deep learning model for image classification. This is a PyTorch implementation.

Keras 63.9k updated yesterday

Keras is a simple, modular, and extensible high-level Python API for deep learning.

fastai 27.9k updated 27d ago

fastai is a library for deep learning research and production. It's built on top of PyTorch.

Microsoft Cognitive Toolkit 17.6k (archived)

Microsoft Cognitive Toolkit (CNTK) is an open-source deep-learning toolkit.

ONNX 20.5k updated yesterday

Open Neural Network Exchange (ONNX) is an open format built to represent machine learning models.

Hugging Face Transformers 158.4k updated yesterday

State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.

DeepMind Control Suite

A toolkit for developing and comparing reinforcement learning agents.

TensorFlow Models 77.7k updated 8d ago

Models and examples built with TensorFlow.

JAX 35.2k updated today

JAX is an alpha library for high performance machine learning research.

Detectron2 34.3k updated 4d ago

Detectron2 is FAIR's next-generation library for object detection and segmentation.

YOLOv5 57.1k updated 7d ago

YOLOv5 is a state-of-the-art, real-time object detection system.

PyTorch Image Models 36.5k updated 2d ago

PyTorch Image Models (timm) is a collection of pre-trained image classification models.

DeepSpeed 41.9k updated yesterday

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

NVIDIA NeMo 17.0k updated yesterday

NeMo is a conversational AI toolkit built from the ground up for researchers and developers to create and experiment with the world's leading pre-trained conversational AI models.

TensorFlow/TensorFlow 37.5k updated yesterday

An open source machine learning framework for everyone

MMDetection 32.5k updated 1y ago

OpenMMLab detection toolbox and benchmark

MMSegmentation 9.7k updated 1y ago

OpenMMLab semantic segmentation toolbox and benchmark

Latent Diffusion Models 13.9k updated 2y ago

High-resolution image synthesis with latent diffusion models

llama.cpp 99.2k updated yesterday

Port of Facebook's LLaMA model in C/C++

GGML 14.3k updated 9d ago

Tensor library for machine learning

OpenCLIP 13.6k updated 13d ago

An open implementation of the OpenAI CLIP model

Stable Diffusion Web UI 162.0k updated 23d ago

A Gradio web UI for Stable Diffusion.

Stable Diffusion

High-Resolution Image Synthesis with Latent Diffusion Models

InvokeAI

InvokeAI is a leading creative engine for Python, providing tools for AI-powered creativity.

CLIP 32.9k updated 1mo ago

CLIP (Contrastive Language–Image Pre-training)

Microsoft Phi-1

A small, high-quality language model for reasoning and language understanding.

Fairseq 32.2k (archived)

A sequence modeling toolkit.

Detectron 26.4k (archived)

Detectron is a tool for object detection and segmentation.

Nvdiffrast 1.8k updated 3mo ago

Nvdiffrast is a library for differentiable rasterization.

EG3D 3.3k updated 2y ago

Efficient Geometry-aware 3D Generative Adversarial Networks

StyleGAN3 6.9k updated 2y ago

StyleGAN3 is an efficient and high-fidelity generative adversarial network for image synthesis.

StyleGAN2 11.2k updated 1y ago

StyleGAN2 is an efficient and high-fidelity generative adversarial network for image synthesis.

StyleGAN

StyleGAN is a generative adversarial network for image synthesis.

Generative Adversarial Networks

Google's Generative Adversarial Networks implementation.

PyTorchVideo 3.6k updated 2mo ago

A PyTorch Video library for video understanding research.

fastText 26.5k (archived)

FastText is a library for efficient learning of word representations and sentence classification.

BERT 39.9k (archived)

BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding

GPT-2 24.7k updated 1y ago

OpenAI's GPT-2 model implementation.

GPT-3 15.8k (archived)

OpenAI's GPT-3 model implementation.

wav2vec

Raw audio sequence to sequence models.

wav2vec 2.0

The official PyTorch implementation of the wav2vec 2.0 paper.

PyTorch-pretrained-BERT 158.3k updated yesterday

A PyTorch implementation of BERT for pre-trained models.

T5 6.5k updated 2mo ago

Text-to-Text Transfer Transformer

Summarization

OpenAI's summarization models.

M2M-100

A large-coverage multilingual neural machine translation model.

Seq2Seq

Google's Seq2Seq models.

Fast Neural Style

A fast and easy-to-use neural style transfer implementation.

Neural Style 18.3k updated 8y ago

Neural style transfer implementation.

Deepfakes

A deepfake implementation.