search

liesel-devs / liesel

A probabilistic programming framework
https://liesel-project.org
MIT License
38 stars 2 forks source link

Adapt to blackjax 1.0.0 #100

Closed wiep closed 11 months ago

wiep commented 11 months ago

Adapts the code for the hmc and nuts kernels to reflect changes in blackjax 1.0.0.

BlackJax changed their kernel interface. They use now classes with the member function step. I adapted the code such that it requires minimal changes in our code base. However, the readability is decreased and we should tidy it up either in this PR or in a subsequent PR.

jobrachem commented 11 months ago

Awesome, that was so quick! Are you still actively working on this or should I already take a look tomorrow? If so, you can remove the draft status.

wiep commented 11 months ago

thanks :-) that was fast 🚀