Jiajun He

RNE: plug-and-play diffusion inference-time control and energy-based training

Vector Quantization using Gaussian Variational Autoencoder

CREPE: Controlling diffusion with REPlica Exchange

Accelerated Parallel Tempering via Neural Transports

Towards Training One-Step Diffusion Models Without Distillation

FEAT: Free energy Estimators with Adaptive Transport

Getting Free Bits Back from Rotational Symmetries in LLMs

Bidirectional Consistency Models

Progressive Tempering Sampler with Diffusion