RobinWu218 / ToSTLinks
[ICLR 2025 Spotlight] Official Implementation for ToST (Token Statistics Transformer)
☆110Updated 4 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☆76Updated last month
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆116Updated last week
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆185Updated 4 months ago
- Triton implement of bi-directional (non-causal) linear attention☆52Updated 5 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆161Updated last month
- ☆222Updated last month
- [CVPR 2025 Highlight] TinyFusion: Diffusion Transformers Learned Shallow☆130Updated 3 months ago
- ☆174Updated 3 weeks ago
- Transformer-Mamba Diffusion Models☆110Updated last year
- Implementation of the proposed MaskBit from Bytedance AI☆82Updated 8 months ago
- [CVPR] MergeVQ: A Unified Framework for Visual Generation and Representation with Token Merging and Quantization☆36Updated 3 weeks ago
- The official implementation of OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows☆76Updated last month
- The official implementation of TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)☆376Updated 2 weeks ago
- Exploring Diffusion Transformer Designs via Grafting☆45Updated last month
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆144Updated 5 months ago
- Official Implementation of LaViDa: :A Large Diffusion Language Model for Multimodal Understanding☆115Updated 3 weeks ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆137Updated 6 months ago
- ☆88Updated last month
- A More Fair and Comprehensive Comparison between KAN and MLP☆171Updated 11 months ago
- [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching☆107Updated last year
- [CVPR2025] Breaking the Low-Rank Dilemma of Linear Attention☆25Updated 4 months ago
- DDT: Decoupled Diffusion Transformer☆264Updated 2 weeks ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆216Updated last month
- ☆130Updated 3 months ago
- The official implementation of "[MASK] is All You Need"☆121Updated 4 months ago
- Scaling RWKV-Like Architectures for Diffusion Models☆135Updated last year
- ☆82Updated 3 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆102Updated last year
- [ICLR 2025] Source code for paper "A Spark of Vision-Language Intelligence: 2-Dimensional Autoregressive Transformer for Efficient Finegr…☆76Updated 7 months ago
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆38Updated 4 months ago