numpyro
numpyro copied to clipboard
Truncated gamma
PR for issue #969 . Contains initial implementation that performs uniform sampling + inverse CDF of Left/Right/Doubly truncated Gamma. Relies on tensorflow functionality for igammainv function, which is not yet implemented at the lax/jax level (see https://github.com/google/jax/issues/5350).
There is a test that fails, but it is not clear to me if this is purely a numerical issue with the uniform + iCDF sampling, or a larger issue that I missed at the time I implemented things.
@quattro Looking the the PR is is the good shape - just have small comments above. Any chance we can have this in the next numpyro release?
Will try my best. Should have some time closer to Thanksgiving holidays, does that fall before next release schedule?
Absolutely, there is no plan for the release date yet. Thank you!