A Proof of Theorem 1: Normalizing Flow as a special case of DiffFlow Proof. As g(t) 0

Below is the implementation of stochastic adjoint method with torch.autograd.Function in PyTorch. The roles of most helper functions and variables can be informed from their names and comments.