code: def init_state(nest: types.Nest) -> RunningStatisticsState:
      """Initializes the running statistics for the given nested structure."""
      dtype = jnp.float64 if jax.config.jax_enable_x64 else jnp.float32
  
      return RunningStatisticsState(  # pytype: disable=wrong-arg-types  # jax-ndarray
          count=0.0,
          mean=_zeros_like(nest, dtype=dtype),
          summed_variance=_zeros_like(nest, dtype=dtype),
          # Initialize with ones to make sure normalization works correctly
          # in the initial state.
          std=_ones_like(nest, dtype=dtype),
      )
  
imports: import dataclasses
  from typing import Any, Optional, Tuple, Union
  from acme import types
  from acme.utils import tree_utils
  import chex
  import jax
  import jax.numpy
  import numpy
  import tree
label_desc: Use image node_id 5 for calling a global function with example usage: init_state(nest) and returns: RunningStatisticsState