rlouf / mcx

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
https://rlouf.github.io/mcx
Apache License 2.0
325 stars 17 forks source link

Call for use cases and examples #3

Open rlouf opened 4 years ago

rlouf commented 4 years ago

Examples

ˋmcx` is only interesting if it can be used and we need more example to showcase the way the library can be used. Examples are a great way to get acquainted with the library and make a first contribution.

We are particularly looking for examples that use:

Use cases

If there is a use case for which you would consider using mcx but cannot because something is missing, let us know in the comments.

rlouf commented 4 years ago

Some ideas:

In terms of practical examples:

miretchin commented 4 years ago

Statistical rethinking examples would be really great to have! Some of the chapters won't yet be possible because it looks like the required distributions are missing, particularly the discrete distributions Beta Binomial, Gamma Poisson, Zero-Inflated Poisson (or zero-inflated negative binomial), and Ordered Logit / Probit.

rlouf commented 3 years ago

Indeed! I will bump the implementation of discrete distributions so that the first release contains enough functionalities to implement most of the examples from Statistical Rethinking.

elanmart commented 3 years ago

Hi @rlouf , if I wanted to start implementing the Statistical Rethinking examples in mcx, should I go with master branch? Or the compiler-refactor?

rlouf commented 3 years ago

Awesome! compiler-refactor is the branch with the most recent version of the API. By the time you finish it should be merged :)

I started here but there's almost nothing: https://github.com/mcx-ppl/statistical-rethinking-mcx. Note that the sample method of distributions returns a 1-dimension array for now, but will soon be a float (PR on the way).

elanmart commented 3 years ago

@rlouf thanks, I'm actually using the book as intro to bayesian methods, so I was planning to implement stuff myself anyway :)

I started with chapters 2 and 3 today (both are still WIP): https://github.com/elanmart/rethinking-2

rlouf commented 3 years ago

Nice! Let me know if you bump into an issue with MCX, something not working or missing.