29 add more distributions

danielward27 / flowjax

https://danielward27.github.io/flowjax/

MIT License

85 stars 10 forks source link

29 add more distributions #30

Closed Tennessee-Wallaceh closed 1 year ago

Tennessee-Wallaceh commented 1 year ago

Adds a number of other distributions to the package, with necessary tests. I refactored the distribution tests to perform similar tests for a list of distribution classes. Implementations rely on jax.scipy.stats for log pdfs or jax.random for sampling.

Adds:

StandardNormal
Uniform
Cauchy
Gumbel
StudentT

closes #29

danielward27 commented 1 year ago

Thanks for the contribution! There's a few minor things that need changing but I'll do that in a separate pull request.