Becker, M., Huot, M., Ritchie, S., & Smith, C. GenJAX: Probabilistic Programming with Gen, built on top of JAX (Version 0.10.0) [Computer software]. https://github.com/ChiSym/genjax