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"
  ☆393Updated 8 months ago
• NVlabs / DiffiT
  [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
  ☆497Updated 10 months ago
• sail-sg / MDT
  Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
  ☆580Updated last year
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆959Updated last year
• facebookresearch / ToMe
  A method to increase the speed and lower the memory footprint of existing vision transformers.
  ☆1,089Updated last year
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,293Updated 5 months ago
• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆1,166Updated 3 months ago
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆601Updated 9 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆431Updated 8 months ago
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆1,023Updated 5 months ago
• horseee / DeepCache
  [CVPR 2024] DeepCache: Accelerating Diffusion Models for Free
  ☆927Updated last year
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆367Updated 2 weeks ago
• zugexiaodui / torch_flops
  A library for calculating the FLOPs in the forward() process based on torch.fx
  ☆127Updated 5 months ago
• SHI-Labs / Neighborhood-Attention-Transformer
  Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
  ☆1,145Updated last year
• whlzy / FiT
  [ICML 2024 Spotlight] FiT: Flexible Vision Transformer for Diffusion Model
  ☆422Updated 10 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆760Updated 9 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆746Updated last month
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆292Updated 10 months ago
• Anima-Lab / MaskDiT
  Code for Fast Training of Diffusion Models with Masked Transformers
  ☆410Updated last year
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆274Updated 3 weeks ago
• CompVis / zigma
  A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)
  ☆334Updated 5 months ago
• hutaiHang / Faster-Diffusion
  [NeurIPS 2024] Official implementation of "Faster Diffusion: Rethinking the Role of UNet Encoder in Diffusion Models"
  ☆341Updated 5 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆407Updated 10 months ago
• NVlabs / FasterViT
  [ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention
  ☆877Updated last month
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆376Updated last year
• lucidrains / magvit2-pytorch
  Implementation of MagViT2 Tokenizer in Pytorch
  ☆630Updated 8 months ago
• meta-pytorch / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆961Updated 3 weeks ago
• mit-han-lab / distrifuser
  [CVPR 2024 Highlight] DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models
  ☆704Updated 9 months ago
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆229Updated 3 weeks ago
• sony / ctm
  ☆300Updated 11 months ago