code: def hungarian_scan(cost):
      """A scan-based batch version of the hungarian matching."""
  
      def hungarian_fn(_, cost):
          return None, hungarian_single(cost)
  
      _, indices = jax.lax.scan(hungarian_fn, None, cost, unroll=1)
      return indices
  
label_desc: Use image node_id 2 for calling a global function with example usage: hungarian_scan(cost) and returns: indices, None,  hungarian_single