timf34

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

pytorch-styleguide

An unofficial styleguide and best practices summary for PyTorch

pytorch-project-template

Deep Learning project template for PyTorch (multi-gpu training is supported)

U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Pytorch-Project-Template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

allennlp

An open-source NLP research library, built on PyTorch.

RepVGG

RepVGG: Making VGG-style ConvNets Great Again

Semi-Hand-Object

gpt-neo

An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.

sportsfield_release

Code release for WACV 2020, "Optimizing Through Learned Errors for Accurate Sports Field Registration"

fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

TimeSformer

The official pytorch implementation of our paper "Is Space-Time Attention All You Need for Video Understanding?"

aws-iot-device-sdk-python-v2

Next generation AWS IoT Client SDK for Python using the AWS Common Runtime

aws-iot-greengrass-deploy-nvidia-deepstream-on-edge

detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

brain-segmentation-pytorch

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI

imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library

torch2trt

An easy to use PyTorch to TensorRT converter

do-you-even-need-attention

Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)

pytorch-template

PyTorch deep learning projects made easy.

MobileNet-V2

A Complete and Simple Implementation of MobileNet-V2 in PyTorch

SoccerNetv2-DevKit

Development Kit for the SoccerNet Challenge

SoccerNet-code

SoccerNet: A Scalable Dataset for Action Spotting in Soccer Videos

videowalk

Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)

deep-daze

Simple command line tool for text to image generation using OpenAI's CLIP and Siren (Implicit neural representation network). Technique was originally created by https://twitter.com/advadnoun

mmdetection

OpenMMLab Detection Toolbox and Benchmark

SoccerDB

SoccerDB: A Large-Scale Database for Comprehensive Video Understanding

pytorchvideo

A deep learning library for video understanding research.