Thanks for your reply, @sammosummo!
I’m sure that any robust PR implementing new features would be very welcome.
Great! I’m working on some tests.
I just wanted to check that you were aware of PyMC4? One of the aims seems to be better GPU usage via tensorflow. I imagine that a biproduct of this would be better multi-node usage as well. I mention it in case it is helpful .
Ah, yes! PyMC4 looks pretty interesting, I checked it out. However, since this is a very critical research (about COVID-19), PyMC4 didn’t sound appropriate since it points itself as an early-stage project. So our decision was to use a consolidated tool (like PyMC3) instead of a new and under development one. But we want to test PyMC4 in the future because of its GPU capabilities, as you mentioned. Thanks!