TorchEBM: A Composable PyTorch Library for Energy-Based and Transport-Based Generative Models
TorchEBM is a PyTorch library for generative models defined either by a scalar potential or by a transport between densities. Energy-based models, diffusion, flow matching, and Schrödinger bridges are factored into one set of composable primitives: energies and fields, interpolants, couplings, training objectives, samp...