code: def move(self):
      start = time.time()
      res = []
      for node in self.nodes:
          res.append(node.move_params.remote())
      ray.get(res)
  
      print(f"Moved weights to TPU in {time.time() - start:.06}s")
  
file_path: mesh-transformer-jax/mesh_transformer/TPU_cluster.py
imports: import itertools
  import json
  import time
  import ray
  from typing import Callable
  import numpy
  from mesh_transformer.train_actor import NetworkRunner
  from google.cloud import storage
  from smart_open import open
  from func_timeout import func_set_timeout
label_desc: Use image node_id 5 for calling the TPUCluster obj's underlying member method code with example usage: obj.move() without return types