[ICLR2025 Spotlightπ₯] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters
β596Feb 11, 2025Updated last year
Alternatives and similar repositories for TokenFormer
Users that are interested in TokenFormer are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Minimal implementation of TokenFormer for inference and learningβ13Nov 6, 2024Updated last year
- [ECCV2024 Oralπ₯] Official Implementation of "GiT: Towards Generalist Vision Transformer through Universal Language Interface"β365Jan 14, 2025Updated last year
- Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, sparsβ¦β387Dec 12, 2024Updated last year
- Code for BLT research paperβ2,064Nov 3, 2025Updated 10 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"β1,297Jul 6, 2026Updated 2 months ago
- Managed Database hosting by DigitalOcean β’ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAIβ300Jun 3, 2025Updated last year
- Next-Token Prediction is All You Needβ2,448Jan 12, 2026Updated 8 months ago
- Helpful tools and examples for working with flex-attentionβ1,253Updated this week
- π Efficient implementations for emerging model architecturesβ5,792Updated this week
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performanceβ¦β155Apr 7, 2025Updated last year
- FlexAttention w/ FlashAttention3 Supportβ27Oct 5, 2024Updated last year
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruningβ154Feb 25, 2026Updated 7 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkβ1,716Mar 16, 2025Updated last year
- A suite of image and video neural tokenizersβ1,728Feb 11, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform β’ AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- β53Jun 24, 2025Updated last year
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793β460May 13, 2025Updated last year
- β93Aug 18, 2024Updated 2 years ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Modelsβ344Feb 23, 2025Updated last year
- Mamba SSM architectureβ18,869Jul 22, 2026Updated 2 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)β167Apr 13, 2025Updated last year
- PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838β1,952Feb 20, 2026Updated 7 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"β94Oct 30, 2024Updated last year
- GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectionβ1,702Oct 28, 2024Updated last year
- Managed Database hosting by DigitalOcean β’ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- β288Jun 6, 2025Updated last year
- Pretraining and inference code for a large-scale depth-recurrent language modelβ970Dec 29, 2025Updated 8 months ago
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"β8,681May 31, 2024Updated 2 years ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clustersβ135Dec 3, 2024Updated last year
- Repository for Meta Chameleon, a mixed-modal early-fusion foundation model from FAIR.β2,102Jul 29, 2024Updated 2 years ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)β25Jun 6, 2024Updated 2 years ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"β258Jun 6, 2025Updated last year
- β19Dec 4, 2025Updated 9 months ago
- Efficient Triton Kernels for LLM Trainingβ6,634Updated this week
- Managed hosting for WordPress and PHP on Cloudways β’ AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"β1,214Dec 22, 2025Updated 9 months ago
- Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"β563Dec 28, 2024Updated last year
- Muon is Scalable for LLM Trainingβ1,547Aug 3, 2025Updated last year
- Official implementation of Next Block Prediction: Video Generation via Semi-Autoregressive Modelingβ42Feb 12, 2025Updated last year
- β311Apr 23, 2025Updated last year
- Autoregressive Model Beats Diffusion: π¦ Llama for Scalable Image Generationβ1,967Aug 15, 2024Updated 2 years ago
- Code for paper "Patch-Level Training for Large Language Models"β110Nov 10, 2025Updated 10 months ago