Skip to content

BlackJAX 0.3.0

Compare
Choose a tag to compare
@rlouf rlouf released this 22 Nov 16:19
· 442 commits to main since this release

What changed

Breaking changes

To build a HMC or NUTS kernel in 0.2.1 and previous versions one needed to provide a potential_fn function:

kernel = nuts.kernel(potential_fn, step_size, inverse_mass_matrix)

Instead we now ask the users to provide the more commonly used log-probability function:

kernel = nuts.kernel(logprob_fn, step_size, inverse_mass_matrix)

where logprob_fn = lambda x: -potential_fn(*x)

New features

Bugs

  • Missing key splitting in trajectory integration (@wiep #53 )