code: def get_ess(samples):
      return tf.function(
          functools.partial(
              tfp.mcmc.effective_sample_size,
              filter_beyond_positive_pairs=True,
          ),
          autograph=False,
      )(samples).numpy()
  
file_path: probability/spinoffs/inference_gym/inference_gym/tools/get_ground_truth.py
imports: import functools
  import os
  import sys
  from absl import app
  from absl import flags
  import numpy
  import pandas
  import tensorflow.compat.v2
  import tensorflow_probability
  from inference_gym.internal import ground_truth_encoding
  from inference_gym.tools.stan import targets
  from tensorflow.python.util import nest
label_desc: Use image node_id 1 for calling a global function with example usage: get_ess(samples) and returns: tf