Sampling Discrete Latent Discrete Variables in Pyro

A MWE (minimum working example) if possible would be very helpful here.

Also here’s a toy model that might help you.