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

github.com/amyeroberts/accelerate

Problem

Audience

Updates

No founder updates yet. Check back soon.