search

Joshuaalbert / jaxns

Probabilistic Programming and Nested sampling in JAX
https://jaxns.readthedocs.io/
Other
141 stars 9 forks source link

Question about numpyo #79

Closed fountaindive closed 1 year ago

fountaindive commented 1 year ago

Hey,

I just tried to use the latest version of jaxns with numpyro and it doesn't seem to work. I think jaxns.prior_transforms has been refactored in v2 or something?

Are you planning on updating numpyro to work with jaxns v2?

Thanks!

Joshuaalbert commented 1 year ago

This requires the work of Du Phan who maintains that. I will take a look at the code though and see if I can make a pr for it.CheersJoshOn Jan 20, 2023 05:15, Sebastian Khan @.***> wrote: Hey, I just tried to use the latest version of jaxns with numpyro and it doesn't seem to work. I think jaxns.prior_transforms has been refactored in v2 or something? Are you planning on updating numpyro to work with jaxns v2? Thanks!

—Reply to this email directly, view it on GitHub, or unsubscribe.You are receiving this because you are subscribed to this thread.Message ID: @.***>

fountaindive commented 1 year ago

Hi,

Ah OK, I thought you maintained it. Thanks for replying / taking a look. Absolutely no issue if not.

Cheers

fehiepsi commented 1 year ago

I took a look but it is quite complicated now given than jaxns is introducing its own ppl. I would suggest to use jaxns ppl for full integration.

We plan to deprecate the support for old jaxns version in numpyro to avoid confusion.

fountaindive commented 1 year ago

OK thanks very much for the info! Closing this now I guess. All the best!