This is the official PyTorch implementation for PIRF: Physics-Informed Reward Fine-Tuning for Diffusion Models (NeurIPS 2025 AI4Science workshop). This repository builds upon DiffusionPDE and EDM. We ...