Neural Networks — Pyro documentation
docs.pyro.ai