code: def run():
      # set a deterministic seed so that both ranks generate the same random tensor
      Tensor.manual_seed(42)
  
      rank = getenv("RANK")
  
      # loop 3 times to make sure it works with the jit
      for _ in range(3):
          # create a tensor to send
          t = Tensor.randn(SIZE, SIZE)
  
          # send to rank 1
          if rank == 0:
              send_jit(t, 1)
          elif rank == 1:
              t2 = Tensor.empty(SIZE, SIZE)
              recv_jit(t2, 0)
  
          # recv from rank 1
          if rank == 0:
              t2 = Tensor.empty(SIZE, SIZE)
              recv_jit(t2, 1)
          elif rank == 1:
              send_jit(t2, 0)
  
          # check that the received tensor is the same as the sent tensor
          if rank == 0:
              assert np.allclose(
                  t.numpy(), t2.numpy()
              ), f"{t2.numpy()} wasn't equal to {t.numpy()}"
  
      print(f"rank {rank} passed")
  
file_path: tinygrad/test/external/external_test_dist_world.py
imports: from extra import dist
  from tinygrad.jit import TinyJit
  from extra.dist import world
  from tinygrad.helpers import CI, getenv
  from tinygrad.tensor import Tensor
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: run() without return types