Consider using https://num.pyro.ai/en/latest/handlers.html#scope to accomplish this (your call on whether to do now or in a separate PR #653
Description
Consider using https://num.pyro.ai/en/latest/handlers.html#scope to accomplish this (your call on whether to do now or in a separate PR
Originally posted by @dylanhmorris in #644 (comment)