Generative Modeling via Drifting in MLX
☆43Feb 6, 2026Updated 5 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…☆73Feb 13, 2026Updated 5 months ago
- Official PyTorch implementation of "Latent Reasoning in TRMs is Secretly a Policy Improvement Operator" (ICML 2026)☆23May 29, 2026Updated 2 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☆22Jul 20, 2026Updated last week
- Generative Modeling with Flux Matching☆65May 11, 2026Updated 2 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.
- ☆36Nov 22, 2024Updated last year
- Official Pytorch Implementation of "Zero-Shot Off-Policy Learning" (ICML 2026)☆25Feb 16, 2026Updated 5 months ago
- A Python package designed for research in diffusion-based generative modeling☆32May 15, 2026Updated 2 months ago
- Personal PyTorch implementation of "Generative Modeling via Drifting" with Claude☆240Feb 6, 2026Updated 5 months ago
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated 9 months ago
- [ICML 2026] Code for Equilibrium Reasoners: learning attractor dynamics for scalable reasoning☆45Jun 1, 2026Updated last month
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆66Dec 19, 2025Updated 7 months ago
- Generative Modeling by Drifting through the Gradient Flow of Sinkhorn Divergence☆26Jun 16, 2026Updated last month
- ☆26May 11, 2026Updated 2 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Official implementation of Categorical Flow Maps on text.☆67Feb 16, 2026Updated 5 months ago
- An implementation is provided here for the NeurIPS2024 paper "MemoryFormer : Minimize Transformer Computation by Removing Fully-Connected…☆16Mar 24, 2026Updated 4 months ago
- Official implementation of Stackelberg PPO for morphology–control co-design.☆16Mar 17, 2026Updated 4 months ago
- Discrete event simulator built in Rust 🦀☆13Jun 10, 2023Updated 3 years ago
- DeFi risk analysis as an ApeWorX plugin☆15Aug 22, 2023Updated 2 years ago
- We revisit the Platonic Representation Hypothesis using calibrated representational similarity metrics with statistical guarantees.☆36Jun 24, 2026Updated last month
- ☆15Jul 22, 2026Updated last week
- An MLX port of Meta's Coconut reasoning model☆16Sep 2, 2025Updated 10 months ago
- This repository consists of two-step training DeepONet code.☆13May 22, 2024Updated 2 years 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.
- vibevoice real time 0.5B swift port☆31Dec 12, 2025Updated 7 months ago
- [ICML 2026] Scaling Beyond Masked Diffusion Language Models☆31Jul 3, 2026Updated 3 weeks ago
- Learn from Your Mistakes: Self-Correcting Masked Diffusion Models☆16Jun 25, 2026Updated last month
- An MLX implementation of Meta AI's ESM-2 protein language model☆16Aug 16, 2025Updated 11 months ago
- JAX implementation of configurable LLM distillation training☆24Nov 15, 2025Updated 8 months ago
- ☆22Jan 23, 2026Updated 6 months ago
- ☆62Jul 22, 2026Updated last week
- d2: Improved Techinques for Training Reasonoing Diffusion Language Models☆16Mar 25, 2026Updated 4 months ago
- Smart contracts of the DFMM protocol☆22Apr 30, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Mythic☆24Oct 15, 2024Updated last year
- Repository for "Training Language Models To Explain Their Own Computations"☆23Jul 7, 2026Updated 3 weeks ago
- Implementation for IceCache: Memory-Efficient KV-cache Management for Long-Sequence LLMs (ICLR 2026).☆20Jun 9, 2026Updated last month
- ☆30Oct 24, 2025Updated 9 months ago
- assign color hues to a collection of text fragments based on embeddings☆20Jun 15, 2024Updated 2 years ago
- (NeurIPS 2023) Residual Q-Learning: Offline and Online Policy Customization without Value☆35Mar 29, 2024Updated 2 years ago
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago