3 ms·
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 you
by jeremiecoullon 6y ago
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/numpyro https://github.com/pyro-ppl/numpyro