code: def softmax(x):
      e = np.exp(x - np.max(x))
      return e / e.sum(axis=-1).reshape([x.shape[0], 1])
  
file_path: NeMo/scripts/asr_language_modeling/ngram_lm/kenlm_utils.py
imports: import gzip
  import json
  import os
  import numpy
  import torch
  from joblib import Parallel, delayed
  from tqdm.auto import tqdm
  import nemo.collections.asr
  from nemo.collections.asr.parts.submodules.ctc_beam_decoding import DEFAULT_TOKEN_OFFSET
  from nemo.utils import logging
label_desc: Use image node_id 1 for calling a global function with example usage: softmax(x) and returns: unknown