pymc.step_methods.hmc.NUTS.fork

NUTS.fork(rng)

Return an independent copy of this step for a new chain/thread.

Shares the compiled logp/dlogp code and read-only data, but gives the copy its own function storage, its own extra_vars shared variables, and fresh adaptation state (mass matrix, step size). Intended to be called before tuning, so the deep-copied adaptation state starts from its initial values.