def leapfrog_one_step(*args): return _leapfrog_integrator_one_step( lambda x: self._log_gamma_log_prob(x, event_dims), independent_chain_ndims, [step_size], *args)
def leapfrog_one_step(*args): return _leapfrog_integrator_one_step( lambda x: self._log_gamma_log_prob(x, event_dims), independent_chain_ndims, [step_size], *args)