parameters: (logpsi: Callable, pars: PyTree, σ: Array, args: PyTree):
return_statement: return jnp.abs(local_value_kernel(logpsi, pars, σ, args)) ** 2
docstring: local_value kernel for MCState and Squared (generic) operators
docstring_summary: local_value kernel for MCState and Squared (generic) operators
docstring_tokens: local_value kernel for MCState and Squared ( generic ) operators
function: def local_value_squared_kernel(logpsi: Callable, pars: PyTree, σ: Array, args: PyTree):
      """
      local_value kernel for MCState and Squared (generic) operators
      """
      return jnp.abs(local_value_kernel(logpsi, pars, σ, args)) ** 2
function_tokens: def local_value_squared_kernel ( logpsi : Callable , pars : PyTree , σ: rray, rgs: yTree) : return jnp . abs ( local_value_kernel ( logpsi , pars , σ, rgs) ) * 