Implementation of Action Matching for the Schrödinger equation

Learning unknown ODE models with Gaussian processes

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations with Finite Element Networks" at ICLR 2022

GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134

PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.

Nonparametric Differential Equation Modeling

Global Waddington-OT

Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022