code: def read_shard(ckpt_dir):
      out = []
      for idx in range(16):
          file_path = ckpt_dir + f"{idx}.npz"
          with open(file_path, "rb") as f:
              buf = f.read()
              f_io = io.BytesIO(buf)
              deserialized = np.load(f_io)
              for i in deserialized:
                  out.append(deserialized[i])
      return out
  
imports: import functools
  import io
  import json
  import time
  import jax
  import jax.numpy
  import numpy
  import multiprocessing
  import ray
  from smart_open import open
  from mesh_transformer.util import head_print
label_desc: Use image node_id 6 for calling a global function with example usage: read_shard(ckpt_dir) and returns: out