Thanks for the help @aseyboldt and @ricardoV94. Fortunately I am still able to sample with numpyro for the moment. I’ll be sure to check this post if a work around becomes available!
Thanks for the help @aseyboldt and @ricardoV94. Fortunately I am still able to sample with numpyro for the moment. I’ll be sure to check this post if a work around becomes available!