Flow Matching Python Packages

rectified-flow-pytorch

Implementation of rectified flow and some of its followup research / improvements in Pytorch

pi-zero-pytorch

Implementation of π₀, the robotic foundation model architecture proposed by Physical Intelligence

f5-tts-mlx

Implementation of F5-TTS in MLX

transfusion-pytorch

Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI

gaia2-pytorch

Implementation of the world model architecture for self driving out of Wayve

gensbi-examples

Examples for the GenSBI library

mintflow

Generation of disentangled microenvironment-induced and intrinsic gene expression vectors from spatial transcriptomics data

gensbi

Generative Models for Simulation-Based Inference in JAX

e2tts-mlx

Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS in MLX

flowmatching-bdt

Implementation of flow matching on tabular data using XGBoost

mlx-e2-tts

Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in MLX

symupe

SyMuPe: Symbolic Music Performance modeling framework

mixture-of-flows

This work introduces Flow Matching Mixture of Experts (FM-MoE), a framework that replaces conventional MLP experts with flow matching networks. Each expert learns a continuous transformation through an ordinary differential equation (ODE), enabling more expressive feature mappings while maintaining the computational efficiency of sparse experts

multicom-ligand

Comprehensive ensembling of protein-ligand structure and affinity prediction methods (CASP16)

inigen

Generation of disentangled microenvironment-induced and intrinsic gene expression vectors from spatial transcriptomics data