Oh, that was not to supply kwargs but for my ad-hoc save-to-disk and peek-while-you-sample framework. I need to have a name for step instance in the scope of my module in order to pickle it
Maybe I’ll add a PR later today to have pm.sample optionally spit out step instance along with the trace. That would enable peeking during sampling for compound steps as well.