littlemcmc.integration.CpuLeapfrogIntegrator

class littlemcmc.integration.CpuLeapfrogIntegrator(potential: littlemcmc.quadpotential.QuadPotential, logp_dlogp_func: Callable[[numpy.ndarray], Tuple[numpy.ndarray, numpy.ndarray]])

Leapfrog integrator using the CPU.

__init__(potential: littlemcmc.quadpotential.QuadPotential, logp_dlogp_func: Callable[[numpy.ndarray], Tuple[numpy.ndarray, numpy.ndarray]]) → None

Instantiate a CPU leapfrog integrator.

Parameters:
potential
logp_dlogp_func

Methods

__init__(potential, logp_dlogp_func, …) Instantiate a CPU leapfrog integrator.
compute_state(q, p) Compute Hamiltonian functions using a position and momentum.
step(epsilon, state[, out]) Leapfrog integrator step.