I sampled 1 chain of 2000 draws in 10 minutes on colab, so again, I think something is wrong with your code.