(2)The Loss function defined in @UCIrvine paper also appears to become infinite while we ensure the loss is finite and results in valid Langevin dynamics.

→ View original post on X — @animaanandkumar