"kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch.distributions</span> <span class="kn">import</span> <span class="p">(</span>
    <span class="n">Uniform</span><span class="p">,</span> 
    <span class="n">Categorical</span><span class="p">,</span> 
    <span class="n">Poisson</span>
<span class="p">)</span>
<span class="kn">from</span> <span class="nn">sbi.utils</span> <span class="kn">import</span> <span class="n">MultipleIndependent</span>
<span class="