RobinWu218 / ToSTLinks
[ICLR 2025 Spotlight] Official Implementation for ToST (Token Statistics Transformer)
☆103Updated 4 months ago
Alternatives and similar repositories for ToST
Users that are interested in ToST are comparing it to the libraries listed below
Sorting:
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆108Updated last month
- [ICML 2025] Fourier Position Embedding: Enhancing Attention’s Periodic Extension for Length Generalization☆71Updated 3 weeks ago
- ☆152Updated last week
- Triton implement of bi-directional (non-causal) linear attention☆50Updated 4 months ago
- Implementation of the proposed MaskBit from Bytedance AI☆82Updated 7 months ago
- The official implementation of OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows☆74Updated last week
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆174Updated 3 months ago
- Official Implementation of LaViDa: :A Large Diffusion Language Model for Multimodal Understanding☆96Updated last week
- Implementation of a multimodal diffusion transformer in Pytorch☆102Updated last year
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆173Updated last year
- A repository for DenseSSMs☆87Updated last year
- [EMNLP 2024] RWKV-CLIP: A Robust Vision-Language Representation Learner☆136Updated last month
- ☆124Updated 11 months ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆137Updated 5 months ago
- Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch☆88Updated 4 months ago
- ☆216Updated 2 weeks ago
- Code for the paper "Vamba: Understanding Hour-Long Videos with Hybrid Mamba-Transformers"☆71Updated 3 months ago
- [CVPR2025] Breaking the Low-Rank Dilemma of Linear Attention☆22Updated 3 months ago
- Scaling Diffusion Transformers with Mixture of Experts☆339Updated 9 months ago
- Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch☆52Updated 7 months ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆138Updated 4 months ago
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆105Updated 3 weeks ago
- N-dimensional Rotary Position Embeddings for PyTorch☆54Updated last year
- Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".☆127Updated 3 weeks ago
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆37Updated 3 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆114Updated 8 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆390Updated 7 months ago
- Transformer-Mamba Diffusion Models☆109Updated 11 months ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆67Updated 2 months ago
- Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".☆68Updated last month