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

Graph Neural Network Library for PyTorch

Must-read papers on graph neural networks (GNN)

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

StyleGAN - Official TensorFlow Implementation

A collaboration friendly studio for NeRFs

LPIPS metric. pip install lpips

A mix of GAN implementations including progressive growing

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

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

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

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

The implementation of StyleGAN on PyTorch 1.0.1

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

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

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