← Search

Leon Klein

8 accepted papers

2025

Amortized Sampling with Transferable Normalizing Flows

NeurIPS 2025poster

Efficient equilibrium sampling of molecular conformations remains a core challenge in computational chemistry and statistical inference. Classical approaches such as molecular dynamics or Markov chain Monte Carlo inherently lack amortization; the computational cost of sampling must be paid in-full f…

Cited by 0SourceScholar
2025

Consistent Sampling and Simulation: Molecular Dynamics with Energy-Based Diffusion Models

NeurIPS 2025poster

In recent years, diffusion models trained on equilibrium molecular distributions have proven effective for sampling biomolecules. Beyond direct sampling, the score of such a model can also be used to derive the forces that act on molecular systems. However, while classical diffusion sampling usually…

Cited by 0SourcecodeScholar
2025

Path Gradients after Flow Matching

NeurIPS 2025poster

Boltzmann Generators have emerged as a promising machine learning tool for generating samples from equilibrium distributions of molecular systems using Normalizing Flows and importance weighting. Recently, Flow Matching has helped speed up Continuous Normalizing Flows (CNFs), scale them to more comp…

Cited by 0SourceScholar
2025

Scalable Equilibrium Sampling with Sequential Boltzmann Generators

ICML 2025poster

Scalable sampling of molecular states in thermodynamic equilibrium is a long-standing challenge in statistical physics. Boltzmann generators tackle this problem by pairing normalizing flows with importance sampling to obtain uncorrelated samples under the target distribution. In this paper, we exten…

Cited by 2SourcePDFScholar
2023

Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamics

NeurIPS 2023spotlight

*Molecular dynamics* (MD) simulation is a widely used technique to simulate molecular systems, most commonly at the all-atom resolution where equations of motion are integrated with timesteps on the order of femtoseconds ($1\textrm{fs}=10^{-15}\textrm{s}$). MD is often used to compute equilibrium p…

2020

Equivariant Flows: Exact Likelihood Generative Learning for Symmetric Densities

ICML 2020poster

Normalizing flows are exact-likelihood generative neural networks which approximately transform samples from a simple prior distribution to samples of the probability distribution of interest. Recent work showed that such generative models can be utilized in statistical mechanics to sample equilibri…

Cited by 287SourcePDFScholar