View source on GitHub |
Makes a Hamiltonian Monte Carlo step function.
oryx.experimental.mcmc.hmc(
unnormalized_log_prob: oryx.core.ppl.LogProbFunction
,
num_leapfrog_steps: int = 5,
step_size: float = 0.1
) -> oryx.core.ppl.LogProbFunction