Make forward sampling functions return InferenceData
#5073
+236
−150
Merged