State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

A Collection of Variational Autoencoders (VAE) in PyTorch.

A large-scale face dataset for face parsing, recognition, generation and editing.

Official implementation of MatterGen -- a generative model for inorganic materials design across the periodic table that can be fine-tuned to steer the generation towards a wide range of property c…

Vector Quantized VAEs - PyTorch Implementation

Cosmos-Reason1 models understand the physical common sense and generate appropriate embodied decisions in natural language through long chain-of-thought reasoning processes.

[CVPR 2022] StyleSwin: Transformer-based GAN for High-resolution Image Generation

Mixture density network implemented in PyTorch.

[Official Implementation] Photo-Realistic Monocular Gaze Redirection Using Generative Adversarial Networks, He et.al. ICCV 2019

A large scale benchmark of materials design methods: https://www.nature.com/articles/s41524-024-01259-w