ghliu

generalized-schrodinger-bridge-matching

code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).

uot-fm

Official Repository for "Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation" [ICLR 2024]

SOC-matching

Code release for "Stochastic Optimal Control Matching"

DMSB

Official implementation of Deep Momentum Schrödinger Bridge

wl-mechanics

SBUnfold

mpot

Implementation of Motion Planning via Optimal Transport (MPOT) in PyTorch, NeurIPS 2023.

optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

line_profiler

Line-by-line profiling for Python

riemannian-fm

code for "Riemannian Flow Matching on General Geometries".

jam

Implementation of Action Matching

presentations

UNSB

Official Repository of "Unpaired Image-to-Image Translation via Neural Schrödinger Bridge" (ICLR 2024)

HeatGeo

Embedding with the Heat-geodesic dissimilarity

dsbm-pytorch

PyTorch Implementation of Diffusion Schrodinger Bridge Matching

ddsm

Dirichlet Diffusion Score Model for Biological Sequence Generation.

fast-ode

Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023

infty-diff

Code for the the ICLR 2024 paper "∞-Diff: Infinite Resolution Diffusion with Subsampled Mollified States"

gDDIM

[ICLR'23 Spotlight] gDDIM: analyze and accelerate general diffusion models, isotropic and non-isotropic

ddib

Dual Diffusion Implicit Bridges for Image-to-Image Translation. ICLR 2023.

pfgmpp

Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"

implicit-sinkhorn

riemannian-score-sde

Score-based generative models for compact manifolds

neural-conservation-law

code for "Neural Conservation Laws A Divergence-Free Perspective".

INDM

Official PyTorch implementation for Maximum Likelihood Training of Implicit Nonlinear Diffusion Model (INDM) in NeurIPS 2022.

GANO

ott

Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.

18336

18.336 - Fast Methods for Partial Differential and Integral Equations

unbiased-teacher-v2

PyTorch code for CVPR 2022 paper Unbiased Teacher v2 Semi-supervised Object Detection for Anchor-free and Anchor-based Detectors

diffae

Official implementation of Diffusion Autoencoders