[CVPR 2026] Denoising, Fast and Slow: Difficulty-Aware Adaptive Sampling for Image Generation

[NeurIPS 2025] DisMo: DIsentangled Motion Representations for Open-World Motion Transfer

[ICCV 2025] SCFlow: Implicitly Learning Style and Content Disentanglement with Flow Models

[CVPR 2026] A PyTorch implementation of the paper "EDGS: Eliminating Densification for Efficient Convergence of 3DGS"

Implementation of the sparse attention pattern proposed by the Deepseek team in their "Native Sparse Attention" paper

Official Implementation for Diffusion Models Without Classifier-free Guidance

Latest Weight Averaging (NeurIPS HITY 2022)

Train VAE like a boss

Implementation of Autoregressive Diffusion in Pytorch

code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"

🚀 Efficient implementations for emerging model architectures

A PyTorch native platform for training generative AI models

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Fast Diffusion Models with Transformers