parameters: (
          self, logpsi: Callable, params: PyTree, x: Array, data: Optional[PyTree]
      )
docstring: r"""This method defines the action of the local operator on a given quantum state
          `logpsi` for a given congfiguration `x`.
          :math:`O_{loc}(x) =  \frac{\bra{x}O{\ket{\psi}}{\bra{x}\ket{\psi}}`
          This method is executed inside of a `jax.jit` block.
          Any static data from the operator itself should be captured in the method.
          Any additional data is provided by the `_pack_arguments`-method
          and will be passed as the `data` argument in this method (Example: masses in kinetic energy).
  
          Args:
              logpsi: variational state
              params: parameters for the variational state
              x: a sample of particle positions
              data: additional data
docstring_summary: r"""This method defines the action of the local operator on a given quantum state
          `logpsi` for a given congfiguration `x`.
          :math:`O_{loc}(x) =  \frac{\bra{x}O{\ket{\psi}}{\bra{x}\ket{\psi}}`
          This method is executed inside of a `jax.jit` block.
          Any static data from the operator itself should be captured in the method.
          Any additional data is provided by the `_pack_arguments`-method
          and will be passed as the `data` argument in this method (Example: masses in kinetic energy).
docstring_tokens: r This method defines the action of the local operator on a given quantum state logpsi for a given congfiguration x . : math : O_ { loc } ( x ) = \ frac { \ bra { x } O { \ ket { \ psi }} { \ bra { x } \ ket { \ psi }} This method is executed inside of a jax . jit block . Any static data from the operator itself should be captured in the method . Any additional data is provided by the _pack_arguments - method and will be passed as the data argument in this method ( Example : masses in kinetic energy ) .
function: def _expect_kernel(
          self, logpsi: Callable, params: PyTree, x: Array, data: Optional[PyTree]
      ):
          r"""This method defines the action of the local operator on a given quantum state
          `logpsi` for a given congfiguration `x`.
          :math:`O_{loc}(x) =  \frac{\bra{x}O{\ket{\psi}}{\bra{x}\ket{\psi}}`
          This method is executed inside of a `jax.jit` block.
          Any static data from the operator itself should be captured in the method.
          Any additional data is provided by the `_pack_arguments`-method
          and will be passed as the `data` argument in this method (Example: masses in kinetic energy).
  
          Args:
              logpsi: variational state
              params: parameters for the variational state
              x: a sample of particle positions
              data: additional data
          """
function_tokens: def _expect_kernel ( self , logpsi : Callable , params : PyTree , x : Array , data : Optional [ PyTree ] ) :