A minimal implementation of Drifting Models for 2D toy data. Unlike diffusion/flow models that iterate at inference, drifting models evolve the pushforward distribution during training and generate in a single forward pass (1-NFE). The drifting field V governs sample movement: V -> 0 as generated matches data.
☆75Feb 13, 2026Updated 6 months ago
Alternatives and similar repositories for Minimal-Drifting-Models
Users that are interested in Minimal-Drifting-Models are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Official PyTorch implementation of "Latent Reasoning in TRMs is Secretly a Policy Improvement Operator" (ICML 2026)☆26May 29, 2026Updated 3 months ago
- ☆20Apr 24, 2026Updated 4 months ago
- PyTorch implementation of Drifting Models by Kaiming He et al.☆21Feb 6, 2026Updated 6 months ago
- Single-step image generation at 306 FPS. Drifting vs Diffusion head-to-head on CIFAR-10.☆46Feb 13, 2026Updated 6 months ago
- Official code base for "General Binding Affinity Guidance for Diffusion Models in Structure-Based Drug Design"☆15Jul 18, 2025Updated last year
- 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.
- ☆17Aug 27, 2026Updated last week
- Generative Modeling by Drifting through the Gradient Flow of Sinkhorn Divergence☆27Jun 16, 2026Updated 2 months ago
- [ICML 2026] Set Diffusion: Interpolating Token Orderings between Autoregression and Diffusion for Fast and Flexible Decoding☆26Aug 9, 2026Updated 3 weeks ago
- Official Pytorch Implementation of "Zero-Shot Off-Policy Learning" (ICML 2026)☆25Feb 16, 2026Updated 6 months ago
- The official repo of the CVPR 2026 paper UniLight☆20Updated this week
- Repository for "Training Language Models To Explain Their Own Computations"☆37Jul 7, 2026Updated last month
- FDFO: Finite Difference Flow Optimization☆117Apr 27, 2026Updated 4 months ago
- ☆15Feb 25, 2026Updated 6 months ago
- ☆12Nov 30, 2023Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Personal PyTorch implementation of "Generative Modeling via Drifting" with Claude☆243Feb 6, 2026Updated 6 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
- Code base for NeurIPS 2022 paper Curriculum Reinforcement Learning using Optimal Transport via Gradual Domain Adaptation.☆11Aug 21, 2023Updated 3 years ago
- Composing Unbalanced Flows for Flexible Docking and Relaxation (ICLR 2025)☆57Jun 19, 2025Updated last year
- ☆15Feb 10, 2026Updated 6 months ago
- H-Net Dynamic Hierarchical Architecture☆81Sep 11, 2025Updated 11 months ago
- Generative Modeling with Flux Matching☆70May 11, 2026Updated 3 months ago
- Global CoT Analysis: Initial attempts to uncover patterns across many chains of thought☆20Feb 10, 2026Updated 6 months ago
- The code for the Ensemble everything everywhere: Multi-scale aggregation for adversarial robustness paper☆23Nov 8, 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.
- eco4cast library aims to reduce carbon footprint of machine learning models with predictive cloud computing scheduling☆16Aug 26, 2024Updated 2 years ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Sep 23, 2024Updated last year
- The first continuous diffusion language model that rivals discrete counterparts on standard language modeling benchmarks like LM1B and Op…☆91Jun 14, 2026Updated 2 months ago
- ☆27Jul 13, 2026Updated last month
- Official Implementation of "Meta Flow Maps enable scalable reward alignment"☆46May 14, 2026Updated 3 months ago
- Interpreting Learned Search and Planning: Reverse-engineering recurrent convolutional networks (DRC) that play Sokoban☆22Jun 29, 2025Updated last year
- Constrained Diffusion with Trust Sampling☆12Dec 18, 2024Updated last year
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆36Jan 24, 2026Updated 7 months ago
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 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.
- ☆19Dec 3, 2024Updated last year
- Official Implementation of iMF https://arxiv.org/abs/2512.02012☆343Feb 27, 2026Updated 6 months ago
- Bilingual Singing Voice Synthesis☆18Mar 25, 2024Updated 2 years ago
- An MLX implementation of Meta AI's ESM-2 protein language model☆16Aug 16, 2025Updated last year
- Code for "Aligning Optimization Trajectories with Diffusion Models for Constrained Design Generation" @ NeurIPS 2023☆25Oct 12, 2023Updated 2 years ago
- Autoregressive Image Generation☆31Jun 13, 2025Updated last year
- ☆30May 11, 2026Updated 3 months ago