DeepSpeed is a distributed deep-learning optimization library for making training and inference faster and more scalable, especially on multi-GPU systems. It targets long or multimodal inputs, quantization, cached weights and inputs, and workloads scaling to thousands of GPUs. section 5.1.2
Use DeepSpeed when the workload needs:
- Multi-GPU scaling, memory management, gradient checkpointing, or caching. section 5.1.2
- Training and inference acceleration with quantization, while recognizing that accuracy may depend on the configuration. section 5.1.2
- A Linux-oriented setup; Windows has partial support and requires additional inference configuration. section 5.1.2

