← Search

Laurence Illing Midgley

3 accepted papers

2024

Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX

ICLR 2024poster

Open-source reinforcement learning (RL) environments have played a crucial role in driving progress in the development of AI algorithms. In modern RL research, there is a need for simulated environments that are performant, scalable, and modular to enable their utilization in a wider range of potent…

2023

Flow Annealed Importance Sampling Bootstrap

ICLR 2023top-25%

Normalizing flows are tractable density models that can approximate complicated target distributions, e.g. Boltzmann distributions of physical systems. However, current methods for training flows either suffer from mode-seeking behavior, use samples from the target generated beforehand by expensive…

2023

SE(3) Equivariant Augmented Coupling Flows

NeurIPS 2023spotlight

Coupling normalizing flows allow for fast sampling and density evaluation, making them the tool of choice for probabilistic modeling of physical systems. However, the standard coupling architecture precludes endowing flows that operate on the Cartesian coordinates of atoms with the SE(3) and permut…