pytorch

karpathy/micrograd

karpathy/micrograd

April 9, 2020

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API
toandaominh1997/EfficientDet.Pytorch

toandaominh1997/EfficientDet.Pytorch

April 9, 2020

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch
zylo117/Yet-Another-EfficientDet-Pytorch

zylo117/Yet-Another-EfficientDet-Pytorch

April 8, 2020

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
caoscott/SReC

caoscott/SReC

April 6, 2020

PyTorch Implementation of "Lossless Image Compression through Super-Resolution"
facebookresearch/pycls

facebookresearch/pycls

March 28, 2020

Codebase for Image Classification Research, written in PyTorch.
vahidk/EffectivePyTorch

vahidk/EffectivePyTorch

March 28, 2020

PyTorch tutorials and best practices.
thu-ml/tianshou

thu-ml/tianshou

March 25, 2020

An elegant, flexible, and superfast PyTorch deep Reinforcement Learning platform.
visionml/pytracking

visionml/pytracking

March 25, 2020

Visual tracking library based on PyTorch.
szymonmaszke/torchlayers

szymonmaszke/torchlayers

March 24, 2020

Shape inference for PyTorch (like in Keras) + new SoTA layers!
facebookresearch/moco

facebookresearch/moco

March 16, 2020

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
NVIDIA-AI-IOT/torch2trt

NVIDIA-AI-IOT/torch2trt

March 4, 2020

An easy to use PyTorch to TensorRT converter
giddyyupp/ganilla

giddyyupp/ganilla

March 3, 2020

pytorch implementation of GANILLA
traveller59/second.pytorch

traveller59/second.pytorch

February 25, 2020

SECOND for KITTI/NuScenes object detection
AntixK/PyTorch-VAE

AntixK/PyTorch-VAE

February 24, 2020

A Collection of Variational Autoencoders (VAE) in PyTorch.
jettify/pytorch-optimizer

jettify/pytorch-optimizer

February 23, 2020

torch-optimizer – collection of optimizers for Pytorch
cszn/KAIR

cszn/KAIR

February 20, 2020

Image Restoration Toolbox (PyTorch). Training and testing codes for DnCNN, FFDNet, SRMD, DPSR, MSRResNet, ESRGAN, IMDN
ayooshkathuria/pytorch-yolo-v3

ayooshkathuria/pytorch-yolo-v3

February 18, 2020

A PyTorch implementation of the YOLO v3 object detection algorithm
dsgiitr/graph_nets

dsgiitr/graph_nets

February 17, 2020

PyTorch Implementation and Explanation of Graph Representation Learning papers involving DeepWalk, GCN, GraphSAGE, ChebNet & GAT.
iamhankai/ghostnet.pytorch

iamhankai/ghostnet.pytorch

February 17, 2020

[CVPR2020] Surpassing MobileNetV3: “GhostNet: More Features from Cheap Operations”
biubug6/Pytorch_Retinaface

biubug6/Pytorch_Retinaface

February 7, 2020

Retinaface get 80.99% in widerface hard val using mobilenet0.25.
kevinzakka/nca

kevinzakka/nca

February 4, 2020

A PyTorch implementation of Neighbourhood Components Analysis.
KevinMusgrave/pytorch-metric-learning

KevinMusgrave/pytorch-metric-learning

February 1, 2020

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.