Optimal Transport for Machine Learners

3 min read Original article ↗

The Book

This book presents optimal transport as a working language for machine learning. It starts from finite assignments and transport plans, builds the Monge and Kantorovich theories, develops Sinkhorn algorithms and generalized distances, and then uses this geometry to study gradient flows, learning dynamics, and transportation-based generative models.

How to cite

Cite the book

Gabriel Peyré. Optimal Transport for Machine Learners. arXiv:2505.06589 [stat.ML], submitted May 10, 2025; cross-listed in cs.AI and math.OC. DOI: 10.48550/arXiv.2505.06589.

PDE4ML survey visual showing transport and PDE dynamics

Focus survey

PDEs for Machine Learning

A long survey of PDE tools for machine learning, written with an optimal-transport bias. It reorganizes the OT4ML material most relevant to dynamic OT, Wasserstein gradient flows, particle limits, diffusion models, flow matching, mean-field training, and transportation views of modern architectures.

Interactive Book

The web version follows the manuscript and places interactive panels beside the mathematical figures, so the reading flow stays centered on the book.

Interactive Kantorovich coupling panel preview Interactive image and histogram transport panel preview Interactive semidiscrete Laguerre cell panel preview Interactive graph Wasserstein transport panel preview

Figure Notebooks

The figure gallery is a searchable database: filter by concept or book section, inspect thumbnails, open notebooks on GitHub, and launch them in Colab.

Book figure preview for McCann shape interpolation Book figure preview for Sinkhorn plans at several regularization strengths Book figure preview for Wasserstein barycenters Book figure preview for Fokker--Planck gradient-flow representations

Teaching Notebooks

Self-contained notebooks for classroom use and quick experimentation. Each can be opened in GitHub or launched directly in Colab.

Course Slides

Four slide decks provide a lecture-oriented route through the computational OT material.

Monge and Kantorovich course slide preview

Monge and Kantorovich

Entropic regularization course slide preview

Entropic Regularization

Dual and semidiscrete course slide preview

Dual and Semidiscrete

Gradient flow and diffusion models course slide preview

Gradient Flow and Diffusion Models