huggingface/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

语言：Python · GitHub 星标：9,934 · Fork：1,588

在 GitHub 上查看原仓库