I recently found that using numpyroc sampling in pymc will use much more memory than pure numpyro. So I had to use numpyro instead. They also have an example of HSGP. It’s not clean and easy to use, but it’s better than the limited memory I have (8GB).