kuleshov-group / mdlm
Simplified Masked Diffusion Language Model
☆207Updated last week
Related projects ⓘ
Alternatives and complementary repositories for mdlm
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆404Updated 8 months ago
- ☆75Updated last year
- A curated list for awesome discrete diffusion models resources.☆67Updated last week
- Some preliminary explorations of Mamba's context scaling.☆191Updated 9 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆95Updated 6 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆239Updated 10 months ago
- Educational implementation of the Discrete Flow Matching paper☆55Updated 2 months ago
- Understand and test language model architectures on synthetic tasks.☆162Updated 6 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆97Updated 4 months ago
- Minimal Implementation of a D3PM in pytorch☆183Updated 6 months ago
- ☆127Updated 10 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆112Updated 2 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆214Updated 3 months ago
- ☆74Updated 11 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆191Updated 2 weeks ago
- ☆109Updated 8 months ago
- This is the official code release for Bayesian Flow Networks.☆255Updated 4 months ago
- Reparameterized Discrete Diffusion Models for Text Generation☆90Updated last year
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆78Updated 2 months ago
- WIP☆89Updated 3 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆85Updated 6 months ago
- Implementation of Infini-Transformer in Pytorch☆104Updated last month
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆113Updated 7 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆174Updated this week
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆256Updated last week
- ☆132Updated last year
- ☆46Updated 5 months ago
- ☆128Updated this week
- Annotated Flow Matching paper☆135Updated 2 months ago