BlackSamorez/tensor_parallel · AI Repo StatsBLBlackSamorez/tensor_parallel
Automatically split your PyTorch models on multiple GPUs for training & inference
Training & fine-tuningPython MIT 42created 2022-10-08 #deep-learning#machine-learning#natural-language-processing#nlp#python#pytorch#pytorch-transformers