code: def format_bytes(num_bytes) -> str:
      suffix = "B"
      suffixes = ["KB", "MB", "GB", "TB"]
      num_bytes = decimal.Decimal(num_bytes)
      one_thousand = decimal.Decimal(1000)
      while suffixes and num_bytes >= one_thousand:
          num_bytes /= one_thousand
          suffix = suffixes.pop(0)
      return f"{num_bytes:.2f} {suffix}"
  
file_path: dm-haiku/haiku/_src/utils.py
imports: import collections.abc
  from collections.abc import Sequence
  import decimal
  import inspect
  import pprint
  import re
  from typing import Any, TypeVar, Union
  import jax
label_desc: Use image node_id 12 for calling a global function with example usage: format_bytes(num_bytes) and returns: num_bytes, str, suffix