Alternatives and similar repositories for NATTEN

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

• naver-ai / rope-vit
  [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"
  ☆373Updated 7 months ago
• NVlabs / DiffiT
  [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
  ☆496Updated 9 months ago
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,255Updated 4 months ago
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆982Updated 4 months ago
• facebookresearch / ToMe
  A method to increase the speed and lower the memory footprint of existing vision transformers.
  ☆1,083Updated last year
• horseee / DeepCache
  [CVPR 2024] DeepCache: Accelerating Diffusion Models for Free
  ☆917Updated last year
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆938Updated last year
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆601Updated 8 months ago
• sail-sg / MDT
  Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
  ☆575Updated last year
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆1,084Updated 2 months ago
• SHI-Labs / Neighborhood-Attention-Transformer
  Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
  ☆1,138Updated last year
• zugexiaodui / torch_flops
  A library for calculating the FLOPs in the forward() process based on torch.fx
  ☆124Updated 4 months ago
• hutaiHang / Faster-Diffusion
  [NeurIPS 2024] Official implementation of "Faster Diffusion: Rethinking the Role of UNet Encoder in Diffusion Models"
  ☆339Updated 4 months ago
• CompVis / zigma
  A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)
  ☆329Updated 4 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆733Updated 2 weeks ago
• NVlabs / FasterViT
  [ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention
  ☆868Updated 3 weeks ago
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆404Updated last year
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆405Updated 7 months ago
• meta-pytorch / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆924Updated 3 weeks ago
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆356Updated this week
• mit-han-lab / distrifuser
  [CVPR 2024 Highlight] DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models
  ☆700Updated 8 months ago
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆269Updated last month
• Dao-AILab / causal-conv1d
  Causal depthwise conv1d in CUDA, with a PyTorch interface
  ☆541Updated 3 weeks ago
• YuchuanTian / U-DiT
  [NeurIPS 2024] The official code of "U-DiTs: Downsample Tokens in U-Shaped Diffusion Transformers"
  ☆214Updated 10 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆291Updated 9 months ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆603Updated last year
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆750Updated 8 months ago
• Xiuyu-Li / q-diffusion
  [ICCV 2023] Q-Diffusion: Quantizing Diffusion Models.
  ☆353Updated last year
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆360Updated 11 months ago
• lucidrains / magvit2-pytorch
  Implementation of MagViT2 Tokenizer in Pytorch
  ☆624Updated 7 months ago