How to set up a pymc environment on google cloud compute platform?

Questions v3
7

Glad you seem to have gotten it working. To use the GPU, you just have to ensure that jax and numpyro are installed, and to use sample_numpyro_nuts instead of sample to run the model.

1 Like