[NeurIPS 2025] Flow x RL. "ReinFlow: Fine-tuning Flow Policy with Online Reinforcement Learning". Support VLAs e.g., pi0, pi0.5. Fully open-sourced.

(ICASSP 2025, official code)FlowSE: Flow Matching-based Speech Enhancement

Official repository for FlowSE (Interspeech 2025)

unofficial Split Mean Flow Implementation from bytedance

Diffusion-SDPO: Safeguarded Direct Preference Optimization for Diffusion Models

(Interspeech 2025, official code) Speech enhancement based on cascaded two flows

Conditional Flow matching for Image and video

A toy implementation of a diffusion transformer based "world model" trained on the game of pong or doom.

Python library designed to make model experimentation seamless and fast. The goal was simple: treat every component (attention heads, MLPs, MoE layers) as a plug-and-play block so you can focus on the architecture, not the boilerplate.

[IROS 2025] PainDiffusion: Learning to Express Pain - IROS Best Entertainment and Amusement Paper Award

A basic implementation of Flow Matching for unconditional 3D point clouds generation.

matting with flow matting

A PyTorch Library for Distribution Transformation in Generative Modeling​

Official Code for ICLR'26 Work LapFlow

Flow Matching-enabled Test-Time Refinement for Unsupervised Cardiac MR Registration

Implicit Dynamical Flow Fusion (IDFF) for Generative Modeling

Implementation of Flow Matching to convert standard normal distribution to multi-modal gaussian

A collection of implementations exploring modern AI architectures and foundational models.

Flowformer is a flow matching-based language model

This repository presents a full implementation and comparison of two generative models evaluated using the Fréchet Inception Distance (FID) on the CIFAR‑10 dataset