Tutorials

RealNVP coupling layers, change of variables, maximum likelihood training, and sampling in JAX and Equinox.

Conditional coupling layers and conditional base densities for learning Gaussian inverse-problem posteriors.

Continuous-time density transformation with Diffrax, Equinox vector fields, and the instantaneous change-of-variables formula.

A conditional continuous flow for a quantity-of-interest posterior associated with the viscous Burgers equation.

Simulation-free vector-field training along conditional probability paths, implemented with JAX, Equinox, Optax, and Diffrax.