Pyro Discussion Forum
Aug 18, 2025 · Forum For Pyro Developers
GPU available but slow compared to CPU on BNN example
Nov 25, 2021 · Well, It is a VI + NeuraReparametrisation to perform a NUTS. Currently, we have found that the job runs 100% on CPU while GPU memory is activated. it’s as if the CPU was …
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 …
Moving MCMC from CPU to GPU - Pyro Discussion Forum
May 9, 2022 · Hi all, I’ve read a few posts on the forum about how to use GPU for MCMC: Transfer SVI, NUTS and MCMC to GPU (Cuda), How to move MCMC run on GPU to CPU …
Implementation & normalizing flow in matrix normal distribution
Nov 1, 2024 · Hi, I’m working on a model where the likelihood follows a matrix normal distribution, X ~ MN_{n,p} (M, U, V). I’m using conjugate priors: M ~ MN U ~ Inverse Wishart V ~ Inverse …
Icdf for discrete distributions - numpyro - Pyro Discussion Forum
May 23, 2025 · Hi there, I am relatively new to numpyro, and I am exploring a bit with different features. In one scenario, I am using Gaussian copulas to model some variables, one of which …
Potential issue with SVI using init_params argument - numpyro
Nov 17, 2024 · I’m trying to utilize SVI with the AutoDelta guide. I run the optimization for a certain number of steps, and save the final parameters to a dictionary and check if the parameters …
Initialize MCMC with SVI - numpyro - Pyro Discussion Forum
Dec 1, 2024 · There are a couple of similar posts here about this, more so in this thread but I don’t think I have a clear understanding of the solution / proper use of init_to_value. I aim to initialize …
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 …
Understanding chain_method - numpyro - Pyro Discussion Forum
Jun 11, 2023 · First of all, thank you for this great library! I am trying to run a model that has a computationally intensive likelihood function. That is, in the definition of my numpyro model, I …