numpyro

numpyro copied to clipboard

transform between domain and codomain of corr_matrix not stable for init_to_uniform strategy

5

llaurabatt

icdf of beta distribution is not implemented

5

```python >>> import numpyro.distributions as dist >>> d = dist.Beta(1, 5) >>> d.icdf(0.975) Traceback (most recent call last): File "", line 1, in File "/home/ayaka/venv/lib/python3.10/site-packages/numpyro/distributions/distribution.py", line 463, in icdf raise...

ayaka14732

Add generalized Mixture distribution.

2

Following the discussion in #1403, this PR introduces a more general mixture distribution combining arbitrary child distributions (so long as they have similar shapes and supports) with fixed weights.

hessammehr

Question about log-likelihood for model with discrete variables

2

Hi, I understand that I can use `funsor.log_density()` to compute the log-joint probability of a sample from a model whose discrete latent sites can be enumerated. However, I'm not sure...

rafaol

Support enumeration in SVI

3

Currently, enumeration is supported in MCMC but is not yet tested in SVI. It would be nice to support this feature for SVI too.

fehiepsi

`SkewMultivariateNormal` and `SkewMultivariateStudentT`

4

I have implemented versions of both of these: ```py from __future__ import annotations from typing import Union, cast import jax from jax import lax from jax import random import jax.numpy...

colehaus

initialize_model() fails with batch size of 1 for model with discrete variables

1

Hi, I'm having issues using NumPyro's latest version of `initialize_model()` on a model with discrete variables. The model samples random variables within a plate context with a data-dependent size. When...

rafaol

A tutorial for integrating ODEs multiple times

12

Hi, I'm wondering if there is a need to contribute with a tutorial for integrating ODEs multiple times based on different initial conditions. It's also possible to extend the current...

A2P2

Example code for using random_flax_module with flax.linen.BatchNorm (mutable)

5

Hi, I am trying to use random_flax_module on a class that uses flax.linen.BatchNorm that uses mutable parameters. Is there any example on how to use that? Here is my code:...

nirmalya-broad

WIP: GumbelSoftmax / RelaxedOneHotCategoricalStraightThrough

8

Hi all, since it's been a while I thought I should maybe give a sign of life and continue from here. This tries to implement https://github.com/pyro-ppl/numpyro/issues/559. There are still some...

daydreamt