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
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Trouble with Initilization and Posterior Sampling | 13 | 2225 | June 15, 2018 | |
| Draw_values() speed/scaling with transformed variables | 9 | 2006 | November 7, 2019 | |
| Generating Posterior Predictive Samples Over Forecast Period | 13 | 1267 | January 2, 2020 | |
| Excessive memory - Multiple regression | 9 | 2931 | October 24, 2017 | |
| PyTorch backend for PyMC4 | 42 | 25919 | April 18, 2018 |