search

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

Add a handler for managing PyroSample plate contexts #3386

Open eb8680 opened 4 months ago

eb8680 commented 4 months ago

Addresses #3385

This PR contains an initial implementation of a PyroSamplePlateScope handler as suggested in #3385.

Still missing tests and documentation, but I thought I'd put a draft up now to support discussion of #3385.