RobinWu218 / ToSTLinks
[ICLR 2025 Spotlight] Official Implementation for ToST (Token Statistics Transformer)
☆120Updated 8 months ago
Alternatives and similar repositories for ToST
Users that are interested in ToST are comparing it to the libraries listed below
Sorting:
- [ICML 2025] Fourier Position Embedding: Enhancing Attention’s Periodic Extension for Length Generalization☆100Updated 4 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆140Updated 9 months ago
- [NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)☆401Updated this week
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆331Updated last month
- The official implementation of "[MASK] is All You Need"☆125Updated 3 months ago
- [CVPR2025] Breaking the Low-Rank Dilemma of Linear Attention☆32Updated 7 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆181Updated last year
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆131Updated last month
- A repository for DenseSSMs☆89Updated last year
- Official Implementation of LaViDa: :A Large Diffusion Language Model for Multimodal Understanding☆158Updated 3 months ago
- When it comes to optimizers, it's always better to be safe than sorry☆375Updated 3 weeks ago
- Code for the paper "Vamba: Understanding Hour-Long Videos with Hybrid Mamba-Transformers" [ICCV 2025]☆90Updated 2 months ago
- Implementation of the proposed MaskBit from Bytedance AI☆82Updated 11 months ago
- Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch☆94Updated 8 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆168Updated 4 months ago
- [NeurIPS 2025 Oral] Exploring Diffusion Transformer Designs via Grafting☆60Updated 4 months ago
- Awesome list of papers that extend Mamba to various applications.☆138Updated 4 months ago
- Scaling RWKV-Like Architectures for Diffusion Models☆138Updated last year
- Transformer-Mamba Diffusion Models☆116Updated last year
- Autoregressive Image Generation with Randomized Parallel Decoding☆79Updated 6 months ago
- TraceRL & TraDo-8B: Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models☆277Updated this week
- DDT: Decoupled Diffusion Transformer☆294Updated 2 months ago
- Triton implement of bi-directional (non-causal) linear attention☆56Updated 8 months ago
- [CVPR 2025 Highlight] TinyFusion: Diffusion Transformers Learned Shallow☆142Updated 6 months ago
- Official repository of InLine attention (NeurIPS 2024)☆56Updated 10 months ago
- ☆138Updated last year
- Discrete Diffusion Forcing (D2F): dLLMs Can Do Faster-Than-AR Inference☆171Updated last month
- [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination☆102Updated 2 months ago
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆39Updated 7 months ago
- Pytorch implementation for MeanFlow☆203Updated 2 months ago