There are several libraries. PyMC4 [1], the next version of PyMC3, will introduce TF as a backend. Tensorflow probability [2] from Google. PYRO [3], from Uber,

There are several libraries. PyMC4 [1], the next version of PyMC3, will introduce TF as a backend. Tensorflow probability [2] from Google. PYRO [3], from Uber, uses Pytorch backend. [1]: https://github.com/pymc-devs/pymc4 https://github.com/pymc-devs/pymc4 [2]: https://www.tensorflow.org/probability/ https://www.tensorflow.org/probability/ [3]: http://pyro.ai/ http://pyro.ai/