
Mixture model with discrete data in Numpyro - Pyro Discussion …
Mar 6, 2024 · Hi. Apologies for the rather long post. I am trying to modify the code from the Gaussian mixture model tutorial (I have also pulled bits of code from various other posts on …
numpyro - Pyro Discussion Forum
Jun 3, 2019 · Forum For Pyro Developers
Missing plate statement on batch dimension? - Pyro Discussion …
Mar 4, 2024 · Hello, I’m new to pyro/numpyro and trying to wrap my head around various details, including but not limited to shapes. I’m getting the below error for my model: ValueError: …
Simplex in model become nan - Misc. - Pyro Discussion Forum
Aug 18, 2022 · I’m building a really simple HMM with Gaussian emission. import pyro import pyro.distributions as dist from pyro import poutine from pyro.optim import Adam import torch …
Reducing MCMC memory usage - numpyro - Pyro Discussion Forum
Oct 15, 2023 · I am running NUTS/MCMC (on multiple CPU cores) for a quite large dataset (400k samples) for 4 chains x 2000 steps. mcmc.run actually ran until the end, but then died with an …
ValueError: Incompatible shapes for ... - Pyro Discussion Forum
May 17, 2024 · Hi, i’m trying to learn multiple (5) coefficients for each treatment group. So i set up my code so that for each “treatment” in the “treatment” plate, i draw 5 numbers from a normal …
Scan + Condition + TransformReparam - Pyro Discussion Forum
Oct 18, 2021 · Following the sgt example I set my observed within the scan using handlers.condition, this works well for a standard distribution but fails for a transformed …
Multi GPU num_samples - Pyro Discussion Forum
Jan 10, 2021 · Hello I was wondering if it’s possible to parallelize the number of particles used to estimate the ELBO over multiple GPUs (i.e. num_particles=40, and if we have 4 GPUs, then …
Dataloaders with numpyro SVI? - Pyro Discussion Forum
Jan 3, 2024 · Are there any examples using standard pytorch dataloaders to train a numpyro svi model in batches? In the example below, there are roughly 2.5 svi updates per second which …
Latent categorical - numpyro - Pyro Discussion Forum
Mar 10, 2021 · It seems numpyro can model latent integers using funsor, great. In this example I want to model 2 latent groups using Dirichlet and Categorical. I am struggling with the ii …