|
|
|
|
|
|
by jeremiecoullon
1944 days 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 |
|
|