original_docstring: 
          Given the current `state`, return the next `state` using the given
          transition kernel.
  
          :param state: A `pytree <https://jax.readthedocs.io/en/latest/pytrees.html>`_
              class representing the state for the kernel. For HMC, this is given
              by :data:`~numpyro.infer.hmc.HMCState`. In general, this could be any
              class that supports `getattr`.
          :param model_args: Arguments provided to the model.
          :param model_kwargs: Keyword arguments provided to the model.
          :return: Next `state`.
          
docstring: Given the current `state`, return the next `state` using the given
  transition kernel.
docstring_tokens: Given the current ` state ` return the next ` state ` using the given transition kernel .
code: def sample(self, state, model_args, model_kwargs):
          """
          Given the current `state`, return the next `state` using the given
          transition kernel.
  
          :param state: A `pytree <https://jax.readthedocs.io/en/latest/pytrees.html>`_
              class representing the state for the kernel. For HMC, this is given
              by :data:`~numpyro.infer.hmc.HMCState`. In general, this could be any
              class that supports `getattr`.
          :param model_args: Arguments provided to the model.
          :param model_kwargs: Keyword arguments provided to the model.
          :return: Next `state`.
          """
          raise NotImplementedError
code_tokens: def sample ( self , state , model_args , model_kwargs ) : raise NotImplementedError
short_docstring: Given the current `state`, return the next `state` using the given
  transition kernel.
short_docstring_tokens: Given the current ` state ` return the next ` state ` using the given transition kernel .
comment: """
          Given the current `state`, return the next `state` using the given
          transition kernel.
  
          :param state: A `pytree <https://jax.readthedocs.io/en/latest/pytrees.html>`_
              class representing the state for the kernel. For HMC, this is given
              by :data:`~numpyro.infer.hmc.HMCState`. In general, this could be any
              class that supports `getattr`.
          :param model_args: Arguments provided to the model.
          :param model_kwargs: Keyword arguments provided to the model.
          :return: Next `state`.
          """