gradoptics.inference.rejection_sampling