I want to share a very raw project, “NumPyro Forecast”. It is based on Pyro’s forecasting module Forecasting — Pyro documentation (an actual port to Jax and NumPyro). I plan to extract many of the custom models I have from my blog and other places. If you are curious or have suggestions, feel free to open some issues. I am open to collaborating with probabilistic forecasters 
3 Likes
you might be interested in [2606.16985] Dynestyx: A Probabilistic Programming Library for Dynamical Systems
2 Likes
Thanks for sharing 
!
Edit: I reach out to them Connection NumPyro Forecasting · Issue #264 · BasisResearch/dynestyx · GitHub