Except for Numba, autograd+GPU is what pytorch offers, all similar to working with Numpy. I have limited experience with deep learning in pytorch but it’s definitely very close to the Numpy experience. And it also has the possibility to add external functions. I don’t know how much the pymc3 guts would have to be changed in this case.
2 Likes