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.
☆73Feb 13, 2026Updated 5 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:
- Generative Modeling via Drifting in MLX☆43Feb 6, 2026Updated 5 months ago
- Official implementation of Categorical Flow Maps on text.☆67Feb 16, 2026Updated 5 months ago
- ☆17Feb 9, 2026Updated 5 months ago
- List of diffusion papers accepted in ECCV 2024.☆15Oct 17, 2024Updated last year
- Official PyTorch implementation of "Latent Reasoning in TRMs is Secretly a Policy Improvement Operator" (ICML 2026)☆23May 29, 2026Updated last month
- 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.
- ☆18Apr 24, 2026Updated 3 months ago
- PyTorch implementation of Drifting Models by Kaiming He et al.☆20Feb 6, 2026Updated 5 months ago
- Single-step image generation at 306 FPS. Drifting vs Diffusion head-to-head on CIFAR-10.☆45Feb 13, 2026Updated 5 months ago
- DLER: Doing Length pEnalty Right - Incentivizing More Intelligence per Token via Reinforcement Learning☆17Nov 11, 2025Updated 8 months ago
- ☆15Updated this week
- Official Pytorch Implementation of "Zero-Shot Off-Policy Learning" (ICML 2026)☆25Feb 16, 2026Updated 5 months ago
- The official repo of the CVPR 2026 paper UniLight☆17Updated this week
- Repository for "Training Language Models To Explain Their Own Computations"☆23Jul 7, 2026Updated 2 weeks ago
- FDFO: Finite Difference Flow Optimization☆115Apr 27, 2026Updated 2 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.
- Simple GRPO scripts and configurations.☆59Feb 6, 2025Updated last year
- ☆12Nov 30, 2023Updated 2 years ago
- MaskFlow: Discrete Flows For Flexible and Efficient Long Video Generation☆28Mar 4, 2025Updated last year
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆46Updated this week
- Official implementation of "Breaking the Factorization Barrier in Diffusion Language Models"☆17Mar 27, 2026Updated 3 months ago
- Personal PyTorch implementation of "Generative Modeling via Drifting" with Claude☆240Feb 6, 2026Updated 5 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
- Official Code Repo for "Learning on the Manifold: Unlocking Standard Diffusion Transformers with Representation Encoders" (ECCV 2026))☆19Jul 15, 2026Updated last week
- Code base for NeurIPS 2022 paper Curriculum Reinforcement Learning using Optimal Transport via Gradual Domain Adaptation.☆11Aug 21, 2023Updated 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.
- ☆15May 20, 2026Updated 2 months ago
- Core Library of Discrete Distribution Networks (ICLR 2025)☆15Oct 12, 2025Updated 9 months ago
- Composing Unbalanced Flows for Flexible Docking and Relaxation (ICLR 2025)☆56Jun 19, 2025Updated last year
- ☆14Feb 10, 2026Updated 5 months ago
- H-Net Dynamic Hierarchical Architecture☆81Sep 11, 2025Updated 10 months ago
- Generative Modeling with Flux Matching☆65May 11, 2026Updated 2 months ago
- ☆476Updated this week
- eco4cast library aims to reduce carbon footprint of machine learning models with predictive cloud computing scheduling☆16Aug 26, 2024Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Sep 23, 2024Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- The first continuous diffusion language model that rivals discrete counterparts on standard language modeling benchmarks like LM1B and Op…☆85Jun 14, 2026Updated last month
- Interpreting Learned Search and Planning: Reverse-engineering recurrent convolutional networks (DRC) that play Sokoban☆21Jun 29, 2025Updated last year
- Implementation of ``Actor-Critic Alignment for Offline-to-Online Reinforcement Learning''☆13Oct 12, 2023Updated 2 years ago
- Code for paper: "Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Design"☆75May 10, 2025Updated last year
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆36Jan 24, 2026Updated 6 months ago
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 years ago
- ☆19Dec 3, 2024Updated last year