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

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 of state-of-the-art linear attention models

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