pyro-ppl / numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
https://num.pyro.ai
Apache License 2.0
2.09k stars 227 forks source link

Add further entropy implementations. #1800

Closed tillahoffmann closed 1 month ago

tillahoffmann commented 1 month ago

Add entropy for Independent, Expanded, Delta, and InverseGamma. Also fixes the __repr__ method of SineSkewed.