[maiybe]

As far as PPLs on Python go, I have found that Pyro's documentation and tutorials are quite stellar. Built on a Pytorch backend (now NumPyro has support for JAX), it's by far my favorite PPL to use.

For your bread and butter PPL starter, I'd go for Variational Autoencoders (VAE). Fun to visualize and tweak: http://pyro.ai/examples/vae.html

If that's too much going on, try this basic tutorial on inference: http://pyro.ai/examples/intro_part_ii.html

The full set of examples is here: http://pyro.ai/examples/