code: def count_corpus(token_list):
      """Return a Counter of the tokens in token_list.
  
      Args:
          token_list: list of token lists
      """
      tokens = [tk for tokens in token_list for tk in tokens]
      return collections.Counter(tokens)
  
imports: import collections
  import random
  import re
  import jax
  import jax.numpy
  import tensorflow_datasets
  import objax
  from objax.functional import one_hot
  from objax.zoo.rnn import RNN
label_desc: Use image node_id 2 for calling a global function with example usage: count_corpus(token_list) and returns: collections