RobinWu218 / ToSTLinks
[ICLR 2025 Spotlight] Official Implementation for ToST (Token Statistics Transformer)
☆113Updated 5 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☆84Updated 2 months ago
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆121Updated last month
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆194Updated 4 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆177Updated last year
- Awesome list of papers that extend Mamba to various applications.☆134Updated 2 months ago
- The official implementation of OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows☆82Updated last month
- Implementation of the proposed MaskBit from Bytedance AI☆82Updated 8 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆102Updated last year
- The official implementation of TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)☆380Updated 2 weeks ago
- ☆194Updated this week
- Official PyTorch implementation of FlowMo.☆87Updated 4 months ago
- Autoregressive Image Generation with Randomized Parallel Decoding☆70Updated 4 months ago
- The official implementation of "[MASK] is All You Need"☆122Updated 2 weeks ago
- Official Pytorch implementation of "Vision Transformers Don't Need Trained Registers"☆83Updated last month
- Official Implementation of LaViDa: :A Large Diffusion Language Model for Multimodal Understanding☆127Updated 3 weeks ago
- N-dimensional Rotary Position Embeddings for PyTorch☆65Updated last year
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆139Updated 6 months ago
- Triton implement of bi-directional (non-causal) linear attention☆51Updated 6 months ago
- Implementation of the dynamic chunking mechanism in H-net by Hwang et al. of Carnegie Mellon☆63Updated last week
- Implementation of Autoregressive Diffusion in Pytorch☆399Updated 9 months ago
- DDT: Decoupled Diffusion Transformer☆268Updated last month
- [Preprint] UCGM: Unified Continuous Generative Models☆165Updated 2 months ago
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆39Updated 5 months ago
- [CVPR2025] Breaking the Low-Rank Dilemma of Linear Attention☆25Updated 5 months ago
- Pytorch implementation of MeanFlow on ImageNet and CIFAR10☆193Updated last week
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆131Updated 6 months ago
- A repository for DenseSSMs☆88Updated last year
- Exploring Diffusion Transformer Designs via Grafting☆48Updated last month
- Towards training VQ-VAE models robustly!☆80Updated 3 weeks ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆151Updated 6 months ago