add torchft buck library

Summary: as title

Reviewed By: d4l3k

Differential Revision: D83488418

Author: tushar00jain

torchft/multiprocessing.py

@@ -8,7 +8,7 @@
 
 
 class _MonitoredPipe:
-    def __init__(self, pipe: "Connection[object, object]") -> None:
+    def __init__(self, pipe: "Connection") -> None:
         self._pipe = pipe
 
     def send(self, obj: object) -> None:

torchft/multiprocessing_test.py

@@ -6,11 +6,11 @@
 from torchft.multiprocessing import _MonitoredPipe
 
 
-def pipe_get(q: "Connection[object, object]") -> None:
+def pipe_get(q: "Connection") -> None:
     q.recv()
 
 
-def pipe_put(q: "Connection[object, object]") -> None:
+def pipe_put(q: "Connection") -> None:
     q.recv()
     q.send(1)
 

torchft/process_group.py

@@ -1443,8 +1443,8 @@ def _worker(
         store_addr: str,
         rank: int,
         world_size: int,
-        req_pipe: "Connection[object, object]",
-        future_pipe: "Connection[object, object]",
+        req_pipe: "Connection",
+        future_pipe: "Connection",
         curr_device: int,
     ) -> None:
         try: