numpyro
numpyro copied to clipboard
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
First and foremost **thanks** for the great work on `numpyro`! **The utility function `random_flax_module()` from the `numpyro.contrib.module` seems to be broken.** As a minimal reproducible example for the error I...
The tutorial mentioned in https://github.com/pyro-ppl/numpyro/issues/1450. It extends the existing example https://num.pyro.ai/en/stable/examples/ode.html by includin integration of multiple initial conditions and treating different data imperfections. Please suggest what to fix/improve.
This is an initial bare bones implementation of the [Divide, Conquer, Combine](https://arxiv.org/abs/1910.13324) inference algorithm for programs with stochastic support as discussed in #1697. I have also included a simple example...
In a [package](github.com/dynamicslab/pysindy) that builds regression problems with a variety of solvers, we recently added numpyro and a regularized horseshoe prior. The tests however, take longer than any other method,...
`AutoNormal`, `AutoDelta`, and `AutoGuideList` raise an exception in SVI when the subsample size varies across different `log_density` evaluation. Here is an example reproducing the issue (run on `master`). ```python import...
I was using `numpyro` for a particular application which required an ICDF and CDF for a Bernoulli, however this was not yet implemented. I use Heaviside functions to calculate the...
The action is copied from https://github.com/pyro-ppl/pyro/pull/3326 TODO: - [ ] Generate secret for TEST_PYPI_API_TOKEN - [ ] Test whether we can publish to TestPyPI - [ ] Modify the action...
# Description It would be great to have Normalizing Flow MCMC method. ## Paper reference - [Adaptive Monte Carlo augmented with normalizing flows](https://arxiv.org/abs/2105.12603) - [Efficient Bayesian Sampling Using Normalizing Flows...
Solves #1696 Adds the following: - `entropy` method for: Bernoulli/Categorical/DiscreteUniform/Geometric - `mode` property for: Bernoulli/Categorical/Binomial/Poisson/Geometric - `name` property for all distributions and explanation for NotImplementedError. Current Status: Passes all local...