/tensor_parallel

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

Primary LanguagePythonMIT LicenseMIT

Watchers