search

tlmakinen / powerbox-jax

Jax implementation of Steven Murray's powerbox: https://github.com/steven-murray/powerbox
Apache License 2.0
5 stars 1 forks source link

Adds numpyro for the tracking of latent variables #2

Open EiffL opened 2 years ago

EiffL commented 2 years ago

It's arguable whether you want this or not ^^. But I'm a big fan of Probabilistic Programming Languages, so even if not directly embeded in the library it would be good to be able to have a way of accessing the latent variables of the model

tlmakinen commented 2 years ago

This is a great idea ! For the IMNN applications I was only interested in crunching the fields, but it would be great to have a numpyro branch handy -- I can see what Stephen thinks since we're trying to integrate the differentiable powerbox with the original