I dont have problems with rewriting to state passing style, and I agree with the potential type issues to the PRNG key. As @eric-haibin-lin mentioned, it can be either a unit type (empty tuple), or a new type to be introduced to relay. Would love to hear more from @tqchen, @MarisaKirisame and @jroesch.