How to predict time series with limited data...

Numpyro AR(1) mean switching model sampling incongrouencies...

GARCH models in Numpyro...

JAX with JIT and custom differentiation...

A JAX custom VJP function for multiple input variable does not work for NumPyro/HMC-NUTS...