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

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Github repo with tutorials to fine tune transformers for diff NLP tasks

Train a state-of-the-art yolov3 object detector from scratch!

Generic template to bootstrap your PyTorch project.

(ICML 2020) This repo contains code for our paper "Revisiting Training Strategies and Generalization Performance in Deep Metric Learning" (https://arxiv.org/abs/2002.08473) to facilitate consistent research in the field of Deep Metric Learning.

Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)

PyTorch Implementations For A Series Of Deep Learning-Based Recommendation Models

Interactively inspect module inputs, outputs, parameters, and gradients.

wandbot is a technical support bot for Weights & Biases' AI developer tools that can run in Discord, Slack, ChatGPT and Zendesk

Phoneme Recognition using pre-trained models Wav2vec2, HuBERT and WavLM. Throughout this project, we compared specifically three different self-supervised models, Wav2vec (2019, 2020), HuBERT (2021) and WavLM (2022) pretrained on a corpus of English speech that we will use in various ways to perform phoneme recognition for different languages wi…

Flexible and scalable template based on PyTorch Lightning + Hydra. Efficient workflow and reproducibility for rapid ML experiments.

KUIELAB-MDX-Net got the 2nd place on the Leaderboard A and the 3rd place on the Leaderboard B in the MDX-Challenge ISMIR 2021

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

Easiest way of fine-tuning HuggingFace video classification models

Train Scene Graph Generation for Visual Genome and GQA in PyTorch >= 1.2 with improved zero and few-shot generalization.

Transformers 3rd Edition

Tensorflow 2.3.0 implementation of DeepLabV3-Plus

A multibranch neural net architecture. (PyTorch)

A clean and structured implementation of Transformer with wandb and pytorch-lightning