from tensorflow.contrib.eager.python.examples.l2hmc import l2hmc

    hparams = get_default_hparams()
    energy_fn, _, _ = l2hmc.get_scg_energy_fn()
    dynamics = l2hmc.Dynamics(
        x_dim=hparams.x_dim,
