/tensor_parallel

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

Primary LanguagePythonMIT LicenseMIT

Watchers

No one’s watching this repository yet.