Type annotate `pyro.nn.dense_nn` and `pyro.nn.auto_reg_nn`

pyro-ppl / pyro

Deep universal probabilistic programming with Python and PyTorch

http://pyro.ai

Apache License 2.0

8.58k stars 987 forks source link

Closed ordabayevy closed 8 months ago

ordabayevy commented 8 months ago

This PR completes type annotation of the pyro.nn submodule.