You can set postprocessing_backend='cpu' to get around that problem.
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Pymc, numpyro GPs and Transform RVs memory behaviour | 9 | 896 | December 21, 2022 | |
| Reduce memory requirements on the GPU when sampling with pm.sampling_jax.sample_numpyro_nuts() | 3 | 1249 | March 15, 2023 | |
| MemeoryError when sampling | 2 | 551 | July 12, 2018 | |
| Has anyone had memory issues with Jax/GPU specifically? | 7 | 5732 | March 22, 2023 | |
| Batch process capability for pymc.sampling_jax.sample_numpyro_nuts() with GPU? | 3 | 583 | September 12, 2022 |