Example: Predator-Prey Model — NumPyro documentation