GitHub repository: BlackSamorez/tensor_parallel
Primary language: Python
Description: Automatically split your PyTorch models on multiple GPUs for training & inference