X-PSI: X-ray Pulse Simulation and Inference

Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Filtering Posterior Sampling) as well as its extension FPS-SMC are proposed.

Code for "Model-based Reinforcement Learning for Continuous Control with Posterior Sampling", ICML 2021