chore: import upstream snapshot with attribution

doc/source/cluster/doc_code/simple-trainer.py

@@ -0,0 +1,32 @@
+import os
+import socket
+import sys
+import time
+
+# trainer.py
+from collections import Counter
+
+import ray
+
+num_cpus = int(sys.argv[1])
+
+ray.init(address=os.environ["ip_head"])
+
+print("Nodes in the Ray cluster:")
+print(ray.nodes())
+
+
+@ray.remote
+def f():
+    time.sleep(1)
+    return socket.gethostbyname("localhost")
+
+
+# The following takes one second (assuming that
+# ray was able to access all of the allocated nodes).
+for i in range(60):
+    start = time.time()
+    ip_addresses = ray.get([f.remote() for _ in range(num_cpus)])
+    print(Counter(ip_addresses))
+    end = time.time()
+    print(end - start)