• Get Started
  • Ecosystem
    Tools

    Learn about the tools and frameworks in the PyTorch Ecosystem

    Ecosystem Day - 2021

    See the posters presented at ecosystem day 2021

    Developer Day - 2021

    See the posters presented at developer day 2021

  • Mobile
  • Blog
  • Tutorials
  • Docs
    PyTorch

    torchaudio

    torchtext

    torchvision

    torcharrow

    TorchData

    TorchRec

    TorchServe

    PyTorch on XLA Devices

  • Resources
    About

    Learn about PyTorch’s features and capabilities

    PyTorch Foundation

    Learn more about the PyTorch Foundation.

    Community

    Join the PyTorch developer community to contribute, learn, and get your questions answered.

    Community stories

    Learn how our community solves real, everyday machine learning problems with PyTorch

    Developer Resources

    Find resources and get questions answered

    Events

    Find events, webinars, and podcasts

    Forums

    A place to discuss PyTorch code, issues, install, research

    Models (Beta)

    Discover, publish, and reuse pre-trained models

  • GitHub
  • X

PyTorch Hub
For Researchers

Explore and extend models from the latest cutting edge research.

All
Audio
Generative
Nlp
Scriptable
Vision
Sort
  • GitHub Stars - Lowest
  • GitHub Stars - Highest
X

  • HybridNets

HybridNets - End2End Perception Network

  • 3D ResNet

Resnet Style Video classification networks pretrained on the Kinetics 400 dataset

  • SlowFast

SlowFast networks pretrained on the Kinetics 400 dataset

  • X3D

X3D networks pretrained on the Kinetics 400 dataset

  • YOLOP

YOLOP pretrained on the BDD100K dataset

  • MiDaS

MiDaS models for computing relative depth from a single image.

  • ntsnet

classify birds using this fine-grained image classifier

  • GPUNet

GPUNet is a new family of Convolutional Neural Networks designed to max out the performance of NVIDIA GPU and TensorRT.

  • Once-for-All

Once-for-all (OFA) decouples training and search, and achieves efficient inference across various edge devices and resource constraints.

  • Open-Unmix

Reference implementation for music source separation

  • Silero Speech-To-Text ...

A set of compact enterprise-grade pre-trained STT Models for multiple languages.

  • Silero Text-To-Speech ...

A set of compact enterprise-grade pre-trained TTS Models for multiple languages

  • Silero Voice Activity ...

Pre-trained Voice Activity Detector

  • YOLOv5

YOLOv5 in PyTorch > ONNX > CoreML > TFLite

  • Deeplabv3

DeepLabV3 models with ResNet-50, ResNet-101 and MobileNet-V3 backbones

  • Transformer (NMT)

Transformer models for English-French and English-German translation.

  • ResNext WSL

ResNext models trained with billion scale weakly-supervised data.

  • DCGAN on FashionGen

A simple generative image model for 64x64 images

  • Progressive Growing of...

High-quality image generation of fashion, celebrity faces

  • Semi-supervised and se...

ResNet and ResNext models introduced in the "Billion scale semi-supervised learning for image classification" paper

  • PyTorch-Transformers

PyTorch implementations of popular NLP Transformers

  • U-Net for brain MRI

U-Net with batch normalization for biomedical image segmentation with pretrained weights for abnormality segmentation in brain MRI

  • EfficientNet

EfficientNets are a family of image classification models, which achieve state-of-the-art accuracy, being an order-of-magnitude smaller and faster....

  • ResNet50

ResNet50 model trained with mixed precision using Tensor Cores.

  • ResNeXt101

ResNet with bottleneck 3x3 Convolutions substituted by 3x3 Grouped Convolutions, trained with mixed precision using Tensor Cores.

  • SE-ResNeXt101

ResNeXt with Squeeze-and-Excitation module added, trained with mixed precision using Tensor Cores.

  • SSD

Single Shot MultiBox Detector model for object detection

  • Tacotron 2

The Tacotron 2 model for generating mel spectrograms from text

  • WaveGlow

WaveGlow model for generating speech from mel spectrograms (generated by Tacotron2)

  • RoBERTa

A Robustly Optimized BERT Pretraining Approach

  • AlexNet

The 2012 ImageNet winner achieved a top-5 error of 15.3%, more than 10.8 percentage points lower than that of the runner up.

  • Densenet

Dense Convolutional Network (DenseNet), connects each layer to every other layer in a feed-forward fashion.

  • FCN

Fully-Convolutional Network model with ResNet-50 and ResNet-101 backbones

  • GhostNet

Efficient networks by generating more features from cheap operations

  • GoogLeNet

GoogLeNet was based on a deep convolutional neural network architecture codenamed "Inception" which won ImageNet 2014.

  • HarDNet

Harmonic DenseNet pre-trained on ImageNet

  • IBN-Net

Networks with domain/appearance invariance

  • Inception_v3

Also called GoogleNetv3, a famous ConvNet trained on Imagenet from 2015

  • MEAL_V2

Boosting Tiny and Efficient Models using Knowledge Distillation.

  • MobileNet v2

Efficient networks optimized for speed and memory, with residual blocks

  • ProxylessNAS

Proxylessly specialize CNN architectures for different hardware platforms.

  • ResNeSt

A new ResNet variant.

  • ResNet

Deep residual networks pre-trained on ImageNet

  • ResNext

Next generation ResNets, more efficient and accurate

  • ShuffleNet v2

An efficient ConvNet optimized for speed and memory, pre-trained on Imagenet

  • SNNMLP

Brain-inspired Multilayer Perceptron with Spiking Neurons

  • SqueezeNet

Alexnet-level accuracy with 50x fewer parameters.

  • vgg-nets

Award winning ConvNets from 2014 Imagenet ILSVRC challenge

  • Wide ResNet

Wide Residual Networks

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources
  • PyTorch
  • Get Started
  • Features
  • Ecosystem
  • Blog
  • Contributing
  • Resources
  • Tutorials
  • Docs
  • Discuss
  • GitHub Issues
  • Brand Guidelines
  • Stay up to date

  • Facebook
  • Twitter
  • YouTube
  • LinkedIn
  • PyTorch Podcasts

  • Spotify
  • Apple
  • Google
  • Amazon
  • Terms
  • |
  • Privacy

© Copyright The Linux Foundation. The PyTorch Foundation is a project of The Linux Foundation. For web site terms of use, trademark policy and other policies applicable to The PyTorch Foundation please see www.linuxfoundation.org/policies/. The PyTorch Foundation supports the PyTorch open source project, which has been established as PyTorch Project a Series of LF Projects, LLC. For policies applicable to the PyTorch Project a Series of LF Projects, LLC, please see www.lfprojects.org/policies/.

  • Get Started
  • Ecosystem
    • Tools
    • Ecosystem Day 2021
    • Developer Day 2021
  • Mobile
  • Blog
  • Tutorials
  • Docs
    • PyTorch
    • torchaudio
    • torchtext
    • torchvision
    • torcharrow
    • TorchData
    • TorchRec
    • TorchServe
    • PyTorch on XLA Devices
  • Resources
    • About
    • PyTorch Foundation
    • Community
    • Community stories
    • Developer Resources
    • Events
    • Forum
    • Models (Beta)
  • GitHub

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.