Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

jax.scipy.stats module missing Poisson distribution

See original GitHub issue

Hi. At the moment, the jax.scipy.stats module does not include the Poisson distribution. Poissons would be quite nice to have, and is needed for use in physics libraries such as pyhf.

Are there any plans to add the Poisson? If not, are there specific issues that are keeping it from being added or just that a volunteer from the community hasn’t stepped up to volunteer the time to do so?

cc @lukasheinrich @kratsg

Issue Analytics

State:
Created 4 years ago
Comments:8 (2 by maintainers)

Top GitHub Comments

1reaction

hawkinspcommented, Oct 17, 2019

@matthewfeickert We are happy either way, at the moment. It’s probably not worth opening a PR until you have something that is more or less functionally correct, e.g., some tests passing.

1reaction

hawkinspcommented, Oct 16, 2019

We welcome contributions! I’m not aware of anything preventing a Poisson distribution. Most likely all it needs is a motivated person who wants to add it. If it’s useful to you, perhaps implement it and send a PR!

Top Results From Across the Web

jax.scipy.stats.poisson.pmf - JAX documentation

LAX-backend implementation of scipy.stats. ... arg1 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for ...

scipy.stats.poisson — SciPy v1.9.3 Manual

A Poisson discrete random variable. As an instance of the rv_discrete class, poisson object inherits from it a collection of generic methods (see...

NumPyro documentation - Pyro

NumPyro is a lightweight probabilistic programming library that provides a NumPy backend for Pyro. We rely on JAX.

NumPyro Documentation - Read the Docs

42 Example: Zero-Inflated Poisson regression model ... Python Module Index ... Distributions: The numpyro.distributions module provides distribution classes ...

How can I obtain the mean of a Poisson distribution given the ...

The cumulative distribution function (CDF) for the Poisson ... from scipy.optimize import fsolve from scipy.stats import poisson def ...