Generative Modeling via Drifting in MLX
☆43Feb 6, 2026Updated 8 months ago
Alternatives and similar repositories for mlx-drifting-model
Users that are interested in mlx-drifting-model are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A minimal implementation of Drifting Models for 2D toy data. Unlike diffusion/flow models that iterate at inference, drifting models evo…☆76Feb 13, 2026Updated 8 months ago
- Official PyTorch implementation of "Latent Reasoning in TRMs is Secretly a Policy Improvement Operator" (ICML 2026)☆26May 29, 2026Updated 4 months ago
- A minimal, single-file implementation of the MeanFlow paper on 2D toy examples, with a side-by-side comparison to rectified flow.☆27Jul 4, 2025Updated last year
- [ICML 2026] Set Diffusion: Interpolating Token Orderings between Autoregression and Diffusion for Fast and Flexible Decoding☆27Aug 9, 2026Updated 2 months ago
- Generative Modeling with Flux Matching☆71May 11, 2026Updated 5 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆36Nov 22, 2024Updated last year
- A Python package designed for research in diffusion-based generative modeling☆32Aug 16, 2026Updated 2 months ago
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated 11 months ago
- ☆31May 11, 2026Updated 5 months ago
- Generative Modeling by Drifting through the Gradient Flow of Sinkhorn Divergence☆26Jun 16, 2026Updated 4 months ago
- [ICML 2026] Code for Equilibrium Reasoners: learning attractor dynamics for scalable reasoning☆57Sep 14, 2026Updated last month
- Official implementation of Categorical Flow Maps on text.☆70Feb 16, 2026Updated 8 months ago
- An implementation is provided here for the NeurIPS2024 paper "MemoryFormer : Minimize Transformer Computation by Removing Fully-Connected…☆16Mar 24, 2026Updated 6 months ago
- Official implementation of Stackelberg PPO for morphology–control co-design.☆19Mar 17, 2026Updated 7 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Implementation and explorations into DiscoRL, Discovering state-of-the-art reinforcement learning algorithms, David Silver's last work at…☆22Jun 13, 2026Updated 4 months ago
- ☆13Mar 21, 2024Updated 2 years ago
- Functional JAX middleware using a compute Node abstraction☆22Sep 6, 2026Updated last month
- An MLX port of Meta's Coconut reasoning model☆16Sep 2, 2025Updated last year
- This repository consists of two-step training DeepONet code.☆13May 22, 2024Updated 2 years ago
- ☆17Sep 26, 2026Updated 3 weeks ago
- [ICML 2026] We revisit the Platonic Representation Hypothesis using calibrated representational similarity metrics with statistical guara…☆40Jun 24, 2026Updated 3 months ago
- vibevoice real time 0.5B swift port☆31Dec 12, 2025Updated 10 months ago
- Learn from Your Mistakes: Self-Correcting Masked Diffusion Models☆16Jun 25, 2026Updated 3 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Deep learning chat based on DL4J☆11Mar 31, 2017Updated 9 years ago
- An MLX implementation of Meta AI's ESM-2 protein language model☆16Aug 16, 2025Updated last year
- [ICML 2026] Scaling Beyond Masked Diffusion Language Models☆37Sep 7, 2026Updated last month
- ☆23Updated this week
- PyTorch implementation of "Field Matching: an Electrostatic Paradigm to Generate and Transfer Data" (ICML 2025)☆21Jul 23, 2025Updated last year
- d2: Improved Techinques for Training Reasonoing Diffusion Language Models☆17Mar 25, 2026Updated 6 months ago
- JAX implementation of configurable LLM distillation training☆24Nov 15, 2025Updated 11 months ago
- ☆30Oct 24, 2025Updated 11 months ago
- (NeurIPS 2023) Residual Q-Learning: Offline and Online Policy Customization without Value☆36Mar 29, 2024Updated 2 years ago
- 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.
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- Embedding Inversion via Conditional Masked Diffusion: recover original text from embedding vectors using parallel denoising. Live demo + …☆61Mar 7, 2026Updated 7 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆246Nov 24, 2025Updated 10 months ago
- Experiments Notebook of "Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanism"☆17Apr 30, 2025Updated last year
- Implementation for IceCache: Memory-Efficient KV-cache Management for Long-Sequence LLMs (ICLR 2026).☆25Jun 9, 2026Updated 4 months ago
- Codebase for the paper "Inverting the Bellman Equation: From Q-Values to World Models"☆40Aug 2, 2026Updated 2 months ago
- ☆23Jul 10, 2026Updated 3 months ago