code: def softmax(x, ieee_rounding=False):
      z = x - max(x, 0)
      num = core.exp(z)
      den = sum(num, 0)
      return core.fdiv(num, den, ieee_rounding)
  
imports: from __future__ import annotations
  from ..runtime.jit import jit
  from .None import core, math
label_desc: Use image node_id 3 for calling a global function with example usage: softmax(x, ieee_rounding) and returns: core