We’ve publicly released dynestyx, a new library for probabilistic inference of dynamical systems, built on top of NumPyro.