liesel.goose.nuts#

No U-Turn Sampler (NUTS).

Classes

NUTSKernel(position_keys[, ...])

A NUTS kernel with dual averaging and an inverse mass matrix tuner, implementing the Kernel protocol.

NUTSKernelState(step_size, inverse_mass_matrix)

A dataclass for the state of a NUTSKernel, implementing the DAKernelState protocol.

NUTSTransitionInfo(error_code, ...)