/accelerate

🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers