pyro

https://pyro.ai/

The StatQuest Introduction to PyTorch