How to increase sampling speed with pm.sample

Thank you, I’m having trouble installing numpyro for JAX sampling, I’m on windows, would I need to use Ubuntu?