blackjax
blackjax copied to clipboard
Refactor the potential fun flip in HMC
We internally flip the sign of log_prob_fn
and use potential_fun
instead - this start to create more headache in the code base, we should refactor to use log_prob_fn
directly.
I'm not sure I'm ready to forego the physical system analogy in the code.
Think of it this way: you can get rid of 100s lines of codes
100s? I see max 20?
Even if it doesn't remove many lines of code this is nevertheless a simplifying change. You were right, adding this to the 1.0
milestone as this would be a breaking change for those using the internals.