4 papers
Memory-Guided Unified Hardware Accelerator for Mixed-Precision Scientific Computing
Chuanzhen Wang, Leo Zhang, Eric Liu
Recent hardware acceleration advances have enabled powerful specialized accelerators for finite element computations, spiking neural network inference, and sparse tensor operations…
CREPE: Controlling Diffusion with Replica Exchange
Jiajun He, Paul Jeha, Peter Potaptchik +5
Inference-time control of diffusion models aims to steer model outputs to satisfy new constraints without retraining. Previous approaches have mostly relied on heuristic guidance o…
The Cosine Schedule is Fisher-Rao-Optimal for Masked Discrete Diffusion Models
Leo Zhang, Saifuddin Syed
In this work, we study the problem of choosing the discretisation schedule for sampling from masked discrete diffusion models in terms of the information geometry of the induced pr…
Accelerated Parallel Tempering via Neural Transports
Leo Zhang, Peter Potaptchik, Jiajun He +5
Markov Chain Monte Carlo (MCMC) algorithms are essential tools in computational statistics for sampling from unnormalised probability distributions, but can be fragile when targeti…