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

[2]: https://www.tensorflow.org/probability/

[3]: http://pyro.ai/