numpyro icon indicating copy to clipboard operation
numpyro copied to clipboard

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Results 154 numpyro issues
Sort by recently updated
recently updated
newest added

### Bug Description A part of https://github.com/pyro-ppl/numpyro/issues/1981 During MCMC run, we cache some functions so that later run might be faster. However, this might cause some leaks. ### Steps to...

bug

### Feature Summary Distributions know their support. Transformations know their domain and codomain. As far as I can tell, it should therefore be possible to confirm that a transformation used...

enhancement

There is some dedicated logic to enhance the performance for specific `init_strategy`s. This logic requires that `init_strategy` is a `partial` or a function that returns a `partial`. E.g., I expected,...

enhancement
refactor

in my research domain (healthcare) many outcomes are right-censored (aka survival analysis / time-to-event); the data for an individual patient are basically a tuple of (observed_time, event / censoring indicator)....

enhancement