ParentFull threadjeremiecoullon·Perhaps an alternative to look into: Numpyro [1] has a JAX backend so can be really fast when compiled; and it can run on GPUs. So that might be helpful for your problem with loads of data.[1] https://github.com/pyro-ppl/numpyroView on HN