BlackSamorez/tensor_parallel

Automatically split your PyTorch models on multiple GPUs for training & inference

4 best points · 0 comments · 1 Hacker News post · 652 GitHub stars · View on GitHub

Hacker News discussions

• Make a PyTorch model tensor-parallel in one line of code (4 points, 0 comments, by borzunov, 2023-03-27)