/tensor_parallel

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

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.