As @mlysy pointed out to us, Blackjax models are a bit annoying to get random numbers working with. In particular, our current pattern is not handled by numpyro.infer.initialize_model, as used in #125, and will actually use a CRN implicitly. This should be ameliorated somehow.
Might be helpful to reference how PFJax handles this.