Datasets, Transforms and Models specific to Computer Vision
Support
Quality
Security
License
Reuse
A neural network that transforms a design mock-up into a static website.
Support
Quality
Security
License
Reuse
You like pytorch? You like micrograd? You love tinygrad! ❤️
Support
Quality
Security
License
Reuse
Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
Support
Quality
Security
License
Reuse
Image augmentation for machine learning experiments.
Support
Quality
Security
License
Reuse
StyleGAN - Official TensorFlow Implementation
Support
Quality
Security
License
Reuse
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
Support
Quality
Security
License
Reuse
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
Support
Quality
Security
License
Reuse
A very simple framework for state-of-the-art Natural Language Processing (NLP)
Support
Quality
Security
License
Reuse
kaldi-asr/kaldi is the official location of the Kaldi project.
Support
Quality
Security
License
Reuse
Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
Support
Quality
Security
License
Reuse
🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production
Support
Quality
Security
License
Reuse
Bringing Old Photo Back to Life (CVPR 2020 oral)
Support
Quality
Security
License
Reuse
r
reinforcement-learning-an-introductionby ShangtongZhang
Python 
12413 Version:Current License: Permissive (MIT)
Python Implementation of Reinforcement Learning: An Introduction
Support
Quality
Security
License
Reuse
Python package built to ease deep learning on graph, on top of existing DL frameworks.
Support
Quality
Security
License
Reuse
Graph Neural Network Library for PyTorch
Support
Quality
Security
License
Reuse
Awesome pre-trained models toolkit based on PaddlePaddle. (400+ models including Image, Text, Audio, Video and Cross-Modal with Easy Inference & Serving)
Support
Quality
Security
License
Reuse
Machine learning, in numpy
Support
Quality
Security
License
Reuse
An open-source NLP research library, built on PyTorch.
Support
Quality
Security
License
Reuse
End-to-End Object Detection with Transformers
Support
Quality
Security
License
Reuse
D
DeepLearningExamplesby NVIDIA
Jupyter Notebook 11094 Version:Current
License: No License (No License)
License: No License (No License)State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
Support
Quality
Security
License
Reuse
Style transfer, deep learning, feature transform
Support
Quality
Security
License
Reuse
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
Support
Quality
Security
License
Reuse
Multilingual Sentence & Image Embeddings with BERT
Support
Quality
Security
License
Reuse
A MNIST-like fashion product database. Benchmark :point_down:
Support
Quality
Security
License
Reuse
Decensoring Hentai with Deep Neural Networks
Support
Quality
Security
License
Reuse
TensorFlow CNN for fast style transfer ⚡🖥🎨🖼
Support
Quality
Security
License
Reuse
StyleGAN2 - Official TensorFlow Implementation
Support
Quality
Security
License
Reuse
超轻量级中文ocr，支持竖排文字识别, 支持ncnn、mnn、tnn推理 ( dbnet(1.8M) + crnn(2.5M) + anglenet(378KB)) 总模型仅4.7M
Support
Quality
Security
License
Reuse
深度学习入门教程, 优秀文章, Deep Learning Tutorial
Support
Quality
Security
License
Reuse
PyTorch package for the discrete VAE used for DALL·E.
Support
Quality
Security
License
Reuse
Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
Support
Quality
Security
License
Reuse
为键盘工作者设计的单词记忆与英语肌肉记忆锻炼软件 / Words learning and English muscle memory training software designed for keyboard workers
Support
Quality
Security
License
Reuse
ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
Support
Quality
Security
License
Reuse
YOLOv3 in PyTorch > ONNX > CoreML > TFLite
Support
Quality
Security
License
Reuse
:mag: Haystack is an open source NLP framework to interact with your data using Transformer models and LLMs (GPT-4, ChatGPT and alike). Haystack offers production-ready tools to quickly build complex question answering, semantic search, text generation applications, and more.
Support
Quality
Security
License
Reuse
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
Support
Quality
Security
License
Reuse
Open3D: A Modern Library for 3D Data Processing
Support
Quality
Security
License
Reuse
:notebook:Solutions to Introduction to Algorithms
Support
Quality
Security
License
Reuse
Data-centric declarative deep learning framework
Support
Quality
Security
License
Reuse
High-Resolution 3D Human Digitization from A Single Image.
Support
Quality
Security
License
Reuse
🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐
Support
Quality
Security
License
Reuse
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
Support
Quality
Security
License
Reuse
a
Jupyter Notebook 8487 Version:Current License: Permissive (Apache-2.0)
Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.
Support
Quality
Security
License
Reuse
ML.NET is an open source and cross-platform machine learning framework for .NET.
Support
Quality
Security
License
Reuse
Pre-Training with Whole Word Masking for Chinese BERT（中文BERT-wwm系列模型）
Support
Quality
Security
License
Reuse
Differentiable Computer Vision Library
Support
Quality
Security
License
Reuse
Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.
Support
Quality
Security
License
Reuse
YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
Support
Quality
Security
License
Reuse
Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning.
Support
Quality
Security
License
Reuse
v
visionby pytorch
Datasets, Transforms and Models specific to Computer Vision
Python 14123Updated: 2 y ago License: Permissive (BSD-3-Clause)
14123Updated: 2 y ago License: Permissive (BSD-3-Clause)Support
Quality
Security
License
Reuse
S
Screenshot-to-codeby emilwallner
A neural network that transforms a design mock-up into a static website.
HTML 14035Updated: 2 y ago License: Proprietary (Proprietary)
14035Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
t
tinygradby geohot
You like pytorch? You like micrograd? You love tinygrad! ❤️
Python 14004Updated: 2 y ago License: Permissive (MIT)
14004Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
t
tensor2tensorby tensorflow
Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
Python 13742Updated: 2 y ago License: Permissive (Apache-2.0)
13742Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
i
imgaugby aleju
Image augmentation for machine learning experiments.
Python 13577Updated: 2 y ago License: Permissive (MIT)
13577Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
s
styleganby NVlabs
StyleGAN - Official TensorFlow Implementation
Python 13465Updated: 2 y ago License: Proprietary (Proprietary)
13465Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
h
horovodby horovod
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
Python 13329Updated: 2 y ago License: Proprietary (Proprietary)
13329Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
n
nniby microsoft
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
Python 12981Updated: 2 y ago License: Permissive (MIT)
12981Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
f
flairby flairNLP
A very simple framework for state-of-the-art Natural Language Processing (NLP)
Python 12863Updated: 2 y ago License: Proprietary (Proprietary)
12863Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
k
kaldiby kaldi-asr
kaldi-asr/kaldi is the official location of the Kaldi project.
Shell 12835Updated: 2 y ago License: Proprietary (Proprietary)
12835Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
u
unilmby microsoft
Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
Python 12771Updated: 2 y ago License: Permissive (MIT)
12771Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
T
TTSby coqui-ai
🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production
Python 12468Updated: 2 y ago License: Weak Copyleft (MPL-2.0)
12468Updated: 2 y ago License: Weak Copyleft (MPL-2.0)Support
Quality
Security
License
Reuse
B
Bringing-Old-Photos-Back-to-Lifeby microsoft
Bringing Old Photo Back to Life (CVPR 2020 oral)
Python 12415Updated: 2 y ago License: Permissive (MIT)
12415Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
r
reinforcement-learning-an-introductionby ShangtongZhang
Python Implementation of Reinforcement Learning: An Introduction
Python 12413Updated: 2 y ago License: Permissive (MIT)
12413Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
d
dglby dmlc
Python package built to ease deep learning on graph, on top of existing DL frameworks.
Python 11863Updated: 2 y ago License: Permissive (Apache-2.0)
11863Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
p
pytorch_geometricby rusty1s
Graph Neural Network Library for PyTorch
Python 11858Updated: 4 y ago License: Permissive (MIT)
11858Updated: 4 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
P
PaddleHubby PaddlePaddle
Awesome pre-trained models toolkit based on PaddlePaddle. (400+ models including Image, Text, Audio, Video and Cross-Modal with Easy Inference & Serving)
Python 11818Updated: 2 y ago License: Permissive (Apache-2.0)
11818Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
n
numpy-mlby ddbourgin
Machine learning, in numpy
Python 11766Updated: 2 y ago License: Strong Copyleft (GPL-3.0)
11766Updated: 2 y ago License: Strong Copyleft (GPL-3.0)Support
Quality
Security
License
Reuse
a
allennlpby allenai
An open-source NLP research library, built on PyTorch.
Python 11506Updated: 2 y ago License: Permissive (Apache-2.0)
11506Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
d
detrby facebookresearch
End-to-End Object Detection with Transformers
Python 11172Updated: 2 y ago License: Permissive (Apache-2.0)
11172Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
D
DeepLearningExamplesby NVIDIA
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
Jupyter Notebook 11094Updated: 2 y ago License: No License (No License)
11094Updated: 2 y ago License: No License (No License)Support
Quality
Security
License
Reuse
F
FastPhotoStyleby NVIDIA
Style transfer, deep learning, feature transform
Python 11009Updated: 2 y ago License: Proprietary (Proprietary)
11009Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
S
Swin-Transformerby microsoft
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
Python 10989Updated: 2 y ago License: Permissive (MIT)
10989Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
s
sentence-transformersby UKPLab
Multilingual Sentence & Image Embeddings with BERT
Python 10938Updated: 2 y ago License: Permissive (Apache-2.0)
10938Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
f
fashion-mnistby zalandoresearch
A MNIST-like fashion product database. Benchmark :point_down:
Python 10843Updated: 2 y ago License: Permissive (MIT)
10843Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
D
DeepCreamPyby deeppomf
Decensoring Hentai with Deep Neural Networks
Python 10775Updated: 4 y ago License: Strong Copyleft (AGPL-3.0)
10775Updated: 4 y ago License: Strong Copyleft (AGPL-3.0)Support
Quality
Security
License
Reuse
f
fast-style-transferby lengstrom
TensorFlow CNN for fast style transfer ⚡🖥🎨🖼
Python 10663Updated: 2 y ago License: No License (No License)
10663Updated: 2 y ago License: No License (No License)Support
Quality
Security
License
Reuse
s
stylegan2by NVlabs
StyleGAN2 - Official TensorFlow Implementation
Python 10446Updated: 2 y ago License: Proprietary (Proprietary)
10446Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
c
chineseocr_liteby DayBreak-u
超轻量级中文ocr，支持竖排文字识别, 支持ncnn、mnn、tnn推理 ( dbnet(1.8M) + crnn(2.5M) + anglenet(378KB)) 总模型仅4.7M
C++ 10444Updated: 2 y ago License: Strong Copyleft (GPL-2.0)
10444Updated: 2 y ago License: Strong Copyleft (GPL-2.0)Support
Quality
Security
License
Reuse
D
DeepLearningby Mikoto10032
深度学习入门教程, 优秀文章, Deep Learning Tutorial
Jupyter Notebook 10420Updated: 2 y ago License: Permissive (Apache-2.0)
10420Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
D
DALL-Eby openai
PyTorch package for the discrete VAE used for DALL·E.
Python 10314Updated: 2 y ago License: Proprietary (Proprietary)
10314Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
d
dopamineby google
Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
Jupyter Notebook 10133Updated: 2 y ago License: Permissive (Apache-2.0)
10133Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
q
qwerty-learnerby Kaiyiwing
为键盘工作者设计的单词记忆与英语肌肉记忆锻炼软件 / Words learning and English muscle memory training software designed for keyboard workers
TypeScript 9816Updated: 2 y ago License: Strong Copyleft (GPL-3.0)
9816Updated: 2 y ago License: Strong Copyleft (GPL-3.0)Support
Quality
Security
License
Reuse
o
onnxruntimeby microsoft
ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
C++ 9569Updated: 2 y ago License: Permissive (MIT)
9569Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
y
yolov3by ultralytics
YOLOv3 in PyTorch > ONNX > CoreML > TFLite
Python 9529Updated: 2 y ago License: Strong Copyleft (AGPL-3.0)
9529Updated: 2 y ago License: Strong Copyleft (AGPL-3.0)Support
Quality
Security
License
Reuse
h
haystackby deepset-ai
:mag: Haystack is an open source NLP framework to interact with your data using Transformer models and LLMs (GPT-4, ChatGPT and alike). Haystack offers production-ready tools to quickly build complex question answering, semantic search, text generation applications, and more.
Python 9144Updated: 2 y ago License: Permissive (Apache-2.0)
9144Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
m
maskrcnn-benchmarkby facebookresearch
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
Python 9110Updated: 2 y ago License: Permissive (MIT)
9110Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
O
Open3Dby isl-org
Open3D: A Modern Library for 3D Data Processing
C++ 9014Updated: 2 y ago License: Proprietary (Proprietary)
9014Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
C
CLRSby gzc
:notebook:Solutions to Introduction to Algorithms
C++ 9001Updated: 2 y ago License: Permissive (MIT)
9001Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
l
ludwigby ludwig-ai
Data-centric declarative deep learning framework
Python 8973Updated: 2 y ago License: Permissive (Apache-2.0)
8973Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
p
pifuhdby facebookresearch
High-Resolution 3D Human Digitization from A Single Image.
Python 8971Updated: 2 y ago License: Proprietary (Proprietary)
8971Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
E
External-Attention-pytorchby xmu-xiaoma666
🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐
Python 8889Updated: 2 y ago License: Permissive (MIT)
8889Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
p
pretrained-models.pytorchby Cadene
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
Python 8768Updated: 2 y ago License: Permissive (BSD-3-Clause)
8768Updated: 2 y ago License: Permissive (BSD-3-Clause)Support
Quality
Security
License
Reuse
a
amazon-sagemaker-examplesby aws
Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.
Jupyter Notebook 8487Updated: 2 y ago License: Permissive (Apache-2.0)
8487Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
m
machinelearningby dotnet
ML.NET is an open source and cross-platform machine learning framework for .NET.
C# 8457Updated: 2 y ago License: Permissive (MIT)
8457Updated: 2 y ago License: Permissive (MIT)Support
Quality
Security
License
Reuse
C
Chinese-BERT-wwmby ymcui
Pre-Training with Whole Word Masking for Chinese BERT（中文BERT-wwm系列模型）
Python 8439Updated: 2 y ago License: Permissive (Apache-2.0)
8439Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
k
korniaby kornia
Differentiable Computer Vision Library
Python 8273Updated: 2 y ago License: Permissive (Apache-2.0)
8273Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
v
vid2vidby NVIDIA
Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.
Python 8248Updated: 2 y ago License: Proprietary (Proprietary)
8248Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse
Y
YOLOXby Megvii-BaseDetection
YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
Python 8231Updated: 2 y ago License: Permissive (Apache-2.0)
8231Updated: 2 y ago License: Permissive (Apache-2.0)Support
Quality
Security
License
Reuse
v
vowpal_wabbitby VowpalWabbit
Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning.
C++ 8230Updated: 2 y ago License: Proprietary (Proprietary)
8230Updated: 2 y ago License: Proprietary (Proprietary)Support
Quality
Security
License
Reuse