Alternatives and similar repositories for NATTEN

Users that are interested in NATTEN are comparing it to the libraries listed below

• NVlabs / DiffiT
  [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
  ☆499Updated 11 months ago
• sail-sg / MDT
  Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
  ☆582Updated last year
• naver-ai / rope-vit
  [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"
  ☆402Updated 9 months ago
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,326Updated 6 months ago
• facebookresearch / ToMe
  A method to increase the speed and lower the memory footprint of existing vision transformers.
  ☆1,106Updated last year
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆968Updated last year
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆412Updated last year
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆452Updated 8 months ago
• SHI-Labs / Neighborhood-Attention-Transformer
  Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
  ☆1,146Updated last year
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆611Updated 10 months ago
• NVlabs / FasterViT
  [ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention
  ☆882Updated 2 months ago
• zugexiaodui / torch_flops
  A library for calculating the FLOPs in the forward() process based on torch.fx
  ☆128Updated 6 months ago
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆373Updated last week
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆1,193Updated 3 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆771Updated 9 months ago
• sony / ctm
  ☆303Updated 11 months ago
• CompVis / zigma
  A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)
  ☆336Updated 6 months ago
• hutaiHang / Faster-Diffusion
  [NeurIPS 2024] Official implementation of "Faster Diffusion: Rethinking the Role of UNet Encoder in Diffusion Models"
  ☆343Updated 6 months ago
• ivan-chai / torch-linear-assignment
  Batch computation of the linear assignment problem on GPU.
  ☆94Updated 2 weeks ago
• horseee / DeepCache
  [CVPR 2024] DeepCache: Accelerating Diffusion Models for Free
  ☆930Updated last year
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆1,039Updated 6 months ago
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆276Updated last month
• baofff / U-ViT
  A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".
  ☆1,046Updated 2 years ago
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆385Updated last year
• tianweiy / DMD2
  (NeurIPS 2024 Oral 🔥) Improved Distribution Matching Distillation for Fast Image Synthesis
  ☆958Updated 6 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆753Updated 2 months ago
• Xiuyu-Li / q-diffusion
  [ICCV 2023] Q-Diffusion: Quantizing Diffusion Models.
  ☆356Updated last year
• whlzy / FiT
  [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model
  ☆424Updated 10 months ago
• OpenGVLab / Vision-RWKV
  [ICLR 2025 Spotlight] Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures
  ☆501Updated 7 months ago
• zai-org / Inf-DiT
  Official implementation of Inf-DiT: Upsampling Any-Resolution Image with Memory-Efficient Diffusion Transformer
  ☆430Updated last year