Alternatives and similar repositories for ToST

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

• TsinghuaC3I / Fourier-Position-Embedding
  [ICML 2025] Fourier Position Embedding: Enhancing Attention’s Periodic Extension for Length Generalization
  ☆104Updated 6 months ago
• qiuzh20 / gated_attention
  The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink…
  ☆273Updated 2 months ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆134Updated last month
• tensorgi / TPA
  [NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)
  ☆427Updated last month
• ApexGen-X / MergeVQ
  [CVPR] MergeVQ: A Unified Framework for Visual Generation and Representation with Token Merging and Quantization
  ☆47Updated 4 months ago
• lucidrains / h-net-dynamic-chunking
  Implementation of the dynamic chunking mechanism in H-net by Hwang et al. of Carnegie Mellon
  ☆65Updated 3 months ago
• pengzhangzhi / Awesome-Mamba
  Awesome list of papers that extend Mamba to various applications.
  ☆139Updated 5 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆382Updated 2 months ago
• qhfan / RALA
  [CVPR2025] Breaking the Low-Rank Dilemma of Linear Attention
  ☆35Updated 8 months ago
• fla-org / flash-bidirectional-linear-attention
  Triton implement of bi-directional (non-causal) linear attention
  ☆56Updated 10 months ago
• lucidrains / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆95Updated 9 months ago
• hp-l33 / AiM
  Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"
  ☆141Updated 10 months ago
• HanGuo97 / log-linear-attention
  ☆256Updated 6 months ago
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆322Updated 3 months ago
• kylesargent / FlowMo
  Official PyTorch implementation of FlowMo.
  ☆103Updated 7 months ago
• TIGER-AI-Lab / Vamba
  Code for the paper "Vamba: Understanding Hour-Long Videos with Hybrid Mamba-Transformers" [ICCV 2025]
  ☆95Updated 4 months ago
• zhijie-group / Discrete-Diffusion-Forcing
  Discrete Diffusion Forcing (D2F): dLLMs Can Do Faster-Than-AR Inference
  ☆205Updated 2 months ago
• Gsunshine / py-meanflow
  Pytorch implementation for MeanFlow
  ☆240Updated 4 months ago
• hp-l33 / ARPG
  Autoregressive Image Generation with Randomized Parallel Decoding
  ☆81Updated last month
• keshik6 / grafting
  [NeurIPS 2025 Oral] Exploring Diffusion Transformer Designs via Grafting
  ☆64Updated 5 months ago
• Hprairie / Bi-Mamba2
  A Triton Kernel for incorporating Bi-Directionality in Mamba2
  ☆75Updated 11 months ago
• HaoyiZhu / MeanFlow-PyTorch
  PyTorch re-implementation for MeanFlow
  ☆111Updated 4 months ago
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆166Updated 10 months ago
• LINs-lab / UCGM
  [Preprint] UCGM: Unified Continuous Generative Models
  ☆169Updated 6 months ago
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆181Updated last year
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆106Updated last year
• ShangyuanTong / FreeFlow
  Official PyTorch Implementation of "Flow Map Distillation Without Data"
  ☆82Updated last week
• NVlabs / DDO
  [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination
  ☆106Updated 4 months ago
• transformer-vq / transformer_vq
  ☆200Updated 2 years ago
• Neur-IO / OptVQ
  Towards training VQ-VAE models robustly!
  ☆87Updated 4 months ago