code: def _memory_usage():
      gc.collect()
      return psutil.Process(os.getpid()).memory_info().rss
  
imports: import copy
  import functools
  import gc
  import os
  import time
  from jax import config
  import numpy
  import psutil
  from tensorflow.compat.v2 import test
  from trax import fastmath
  from trax import layers
  from trax import models
  from trax import shapes
  from trax.supervised import decoding
label_desc: Use image node_id 3 for calling a global function with example usage: _memory_usage() and returns: psutil