Hey PyMC:
I’ve been trying colab as a way to improve modelling speed. Does anyone have a working config for colab for state space? (Edit: I’m using a model on colab that I’ve successfully run locally.)
I tried using default nuts sampler and numpyro_nuts and was able to get sampling. But the tree depth goes to 255 after the first 10 samples and then pretty much grinds to a halt. Maybe initial values aren’t being properly set?
I could not get nutpie installed. I got an error about a subprocess failing - and Gemini suggested nutpie needed Rust to compile. Which brought me to conda and the pre-compiled libraries. I also tried to install a conda environment - but was somewhat stymied in the attempt, but I forget why.
I am using the jax backend and set jax to cuda. I used the A100 GPU, and I tried a few others.
Anyway, any help you could provide to this first-timer would be appreciated.
Thank you for your time,
Roy