Alternatives and similar repositories for ToST:

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

• zhixuan-lin / forgetting-transformer
  [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"
  ☆99Updated last month
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆160Updated last month
• hp-l33 / AiM
  Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"
  ☆136Updated 3 months ago
• lucidrains / maskbit-pytorch
  Implementation of the proposed MaskBit from Bytedance AI
  ☆76Updated 5 months ago
• fla-org / flash-bidirectional-linear-attention
  Triton implement of bi-directional (non-causal) linear attention
  ☆47Updated 3 months ago
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆233Updated 3 weeks ago
• jacklishufan / OmniFlows
  The official implementation of OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows
  ☆58Updated last month
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆101Updated 10 months ago
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆113Updated 3 months ago
• mit-han-lab / x-attention
  XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆146Updated last month
• TsinghuaC3I / Fourier-Position-Embedding
  [ICML 2025] Fourier Position Embedding: Enhancing Attention’s Periodic Extension for Length Generalization
  ☆28Updated 3 months ago
• test-time-training / ttt-lm-kernels
  Inference Speed Benchmark for Learning to (Learn at Test Time): RNNs with Expressive Hidden States
  ☆66Updated 9 months ago
• feizc / DiT-MoE
  Scaling Diffusion Transformers with Mixture of Experts
  ☆317Updated 8 months ago
• cfifty / rotation_trick
  ☆98Updated last month
• zbr17 / OptVQ
  Towards training VQ-VAE models robustly!
  ☆72Updated 4 months ago
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆172Updated 10 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆82Updated 2 months ago
• horseee / learning-to-cache
  [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching
  ☆102Updated 9 months ago
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆116Updated 2 months ago
• causalfusion / causalfusion
  ☆159Updated 4 months ago
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆112Updated 6 months ago
• YuqingWang1029 / TokenBridge
  TokenBridge: Bridging Continuous and Discrete Tokens for Autoregressive Visual Generation. https://yuqingwang1029.github.io/TokenBridge
  ☆108Updated this week
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆222Updated last month
• OliverRensu / MVAR
  ☆70Updated 5 months ago
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆88Updated last week
• prathebaselva / FORA
  FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.
  ☆45Updated 10 months ago
• apple / ml-flextok
  FlexTok: Resampling Images into 1D Token Sequences of Flexible Length
  ☆132Updated last month
• End2End-Diffusion / REPA-E
  Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers
  ☆190Updated 3 weeks ago
• philippe-eecs / small-vision
  A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.
  ☆35Updated 10 months ago
• sustcsonglin / linear-attention-and-beyond-slides
  ☆71Updated 2 months ago