Defining Vectorizable Helper Function given Posterior Draws (Forecasting with BVAR)

The issue is the Numba backend does nor support all forms of advanced indexing that the default C backend does. You can try to use the JAX backend with numpyro nuts sampler and see how that goes

1 Like