Notebooks

---

Simple Training/Sampling example

Train and sample from a 2D Flow Matching model.

Discrete Flow Matching

Train and sample from a 2D Discrete Flow Matching model.

Riemannian Flow Matching (Sphere)

2D sphere riemannian flow matching example

Riemannian Flow Matching (Flat Torus)

2D flat torus riemannian flow matching example