All-in-one training for vision models (YOLO, ViTs, RT-DETR, DINOv3): pretraining, fine-tuning, distillation.

Graph Neural Network Library for PyTorch

kaldi-asr/kaldi is the official location of the Kaldi project.

A collaboration friendly studio for NeRFs

Learn how to design, develop, deploy and iterate on production-grade ML applications.

LPIPS metric. pip install lpips

State-of-the-art, simple, fast unbounded / large-scale NeRFs.

[ICCV 2023] MatrixCity: A Large-scale City Dataset for City-scale Neural Rendering and Beyond.

StyleGAN - Official TensorFlow Implementation

Must-read papers on graph neural networks (GNN)

A mix of GAN implementations including progressive growing

Implementation A Style-Based Generator Architecture for Generative Adversarial Networks in PyTorch

Modality-Transferable-MER, multimodal emotion recognition model with zero-shot and few-shot abilities.

Kungfupanda at SemEval-2020 Task 12: BERT-Based Multi-Task Learning for Offensive Language Detection

The implementation of StyleGAN on PyTorch 1.0.1

Code for replication of the paper "The relativistic discriminator: a key element missing from standard GAN"

Implementations of various VAE-based semi-supervised and generative models in PyTorch

VGGVox models for Speaker Identification and Verification trained on the VoxCeleb (1 & 2) datasets