accelerate
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
8k stars
96 watching
937 forks
Language: Python
last commit: 12 days ago
Linked from 2 awesome lists