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
  ☆99Updated 4 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆308Updated 2 weeks ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆132Updated last week
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆159Updated 8 months ago
• keshik6 / grafting
  [NeurIPS 2025 Oral] Exploring Diffusion Transformer Designs via Grafting
  ☆54Updated 3 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 / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆94Updated 7 months ago
• tensorgi / TPA
  [NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)
  ☆394Updated last week
• zhijie-group / Discrete-Diffusion-Forcing
  Discrete Diffusion Forcing (D2F): dLLMs Can Do Faster-Than-AR Inference
  ☆151Updated last week
• fla-org / flash-bidirectional-linear-attention
  Triton implement of bi-directional (non-causal) linear attention
  ☆56Updated 8 months ago
• lucidrains / h-net-dynamic-chunking
  Implementation of the dynamic chunking mechanism in H-net by Hwang et al. of Carnegie Mellon
  ☆65Updated last month
• HanGuo97 / log-linear-attention
  ☆249Updated 3 months ago
• horseee / dKV-Cache
  [NeurIPS'25] dKV-Cache: The Cache for Diffusion Language Models
  ☆101Updated 4 months ago
• TIGER-AI-Lab / Vamba
  Code for the paper "Vamba: Understanding Hour-Long Videos with Hybrid Mamba-Transformers" [ICCV 2025]
  ☆86Updated 2 months ago
• Neur-IO / OptVQ
  Towards training VQ-VAE models robustly!
  ☆85Updated 2 months ago
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆124Updated 2 months ago
• WailordHe / DenseSSM
  A repository for DenseSSMs
  ☆88Updated last year
• hp-l33 / ARPG
  Autoregressive Image Generation with Randomized Parallel Decoding
  ☆77Updated 6 months ago
• NUS-HPC-AI-Lab / Dynamic-Diffusion-Transformer
  ☆87Updated 6 months ago
• Gen-Verse / dLLM-RL
  TraceRL: Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models
  ☆226Updated this week
• LINs-lab / UCGM
  [Preprint] UCGM: Unified Continuous Generative Models
  ☆170Updated 4 months ago
• transformer-vq / transformer_vq
  ☆197Updated last year
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆105Updated last year
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆276Updated last month
• jacklishufan / LaViDa
  Official Implementation of LaViDa: :A Large Diffusion Language Model for Multimodal Understanding
  ☆150Updated 2 months ago
• pengzhangzhi / Awesome-Mamba
  Awesome list of papers that extend Mamba to various applications.
  ☆138Updated 3 months ago
• VainF / TinyFusion
  [CVPR 2025 Highlight] TinyFusion: Diffusion Transformers Learned Shallow
  ☆141Updated 5 months ago
• test-time-training / ttt-lm-kernels
  Inference Speed Benchmark for Learning to (Learn at Test Time): RNNs with Expressive Hidden States
  ☆73Updated last year
• kylesargent / FlowMo
  Official PyTorch implementation of FlowMo.
  ☆95Updated 5 months ago
• Hprairie / Bi-Mamba2
  A Triton Kernel for incorporating Bi-Directionality in Mamba2
  ☆75Updated 9 months ago