License Apache 2.0
PyPI version
Downloads
Build
OpenSSF Best Practices
Twitter
Japanese Twitter
Chinese Zhihu
Slack
<div align="center">
<img src="https://raw.githubusercontent.com/deepspeedai/DeepSpeed/HEAD/docs/assets/images/DeepSpeed_light.svg#gh-light-mode-only" width="400px">
<img src="https://raw.githubusercontent.com/deepspeedai/DeepSpeed/HEAD/docs/assets/images/DeepSpeed_dark_transparent.svg#gh-dark-mode-only" width="400px">
</div>
<p align="center">
<b>English</b> · <a href="README.zh.md">简体中文</a>
</p>
오피스 아워 (Office Hours)
DeepSpeed는 매월 마지막 화요일 America/New_York 시간 12:00에 정기적인 오피스 아워를 개최하며, 이 자리에서 개발 계획이나 기능 등에 대해 논의합니다. 이 회의는 누구나 참여하여 질문할 수 있는 공개 세션입니다.
The meeting은 Zoom을 통해 진행되며 여기에서 참여할 수 있습니다.
최신 소식 (Latest News)
<!-- NOTE: we must use html for news items otherwise links will be broken in the 'more news' section -->
<details>
<!-- NOTE: Maintain only three items in 'more news' section -->
<summary>더 많은 소식 (More news)</summary>
<ul>
<li>[2025/04] <a href="https://github.com/deepspeedai/DeepSpeed/blob/master/blogs/deepcompile/README.md">DeepCompile: 분산 학습을 위한 컴파일러 최적화 잠금 해제</a></li>
<li>[2025/03] <a href="https://github.com/deepspeedai/DeepSpeed/blob/master/blogs/huggingface-tp/README.md">DeepSpeed AutoTP: Hugging Face 모델의 자동 텐서 병렬 학습</a></li>
<li>[2024/12] <a href="https://github.com/deepspeedai/DeepSpeed/blob/master/blogs/ulysses-offload/README.md">Ulysses-Offload: 긴 컨텍스트 LLM 학습의 민주화</a></li>
</ul>
</details>
DL 학습을 위한 극도의 속도와 규모 (Extreme Speed and Scale for DL Training)
***DeepSpeed는 MT-530B(https://www.microsoft.com/en-us/research/blog/using-deepspeed-and-megatron-to-train-megatron-turing-nlg-530b-the-worlds-largest-and-most-powerful-generative-language-model/) 및 BLOOM(https://huggingface.co/blog/bloom-megatron-deepspeed)와 같이 세계에서 가장 강력한 언어 모델을 구현했습니다 (본문 작성 시점 기준). DeepSpeed는 시스템 혁신의 집합체로서, 대규모 DL 학습을 효과적이고 효율적으로 만들었으며, 사용 편의성을 크게 개선하고, 가능한 규모 면에서 DL 학습 환경 자체를 재정의했습니다. 이러한 혁신에는 ZeRO, ZeRO-Infinity, 3D-Parallelism, Ulysses Sequence Parallelism, DeepSpeed-MoE 등이 포함됩니다.
DeepSpeed 채택 (DeepSpeed Adoption)
DeepSpeed는 Microsoft의 AI at Scale 이니셔티브에서 중요한 부분을 차지했습니다. 이 이니셔티브를 통해 사용자는 차세대 AI 기능을 대규모로 구현할 수 있으며, 더 많은 정보는 여기에서 확인할 수 있습니다.
DeepSpeed는 다양한 대규모 모델을 학습하는 데 사용되었으며, 저희가 알고 있는 몇 가지 예시는 다음과 같습니다 (모델 포함을 원하시면 PR을 제출해 주세요):
DeepSpeed는 다음과 같은 여러 인기 오픈소스 DL 프레임워크와 통합되었습니다:
문서 (Documentation)
빌드 파이프라인 상태 (Build Pipeline Status)
설치 (Installation)
DeepSpeed를 시작하는 가장 빠른 방법은 pip을 이용하는 것입니다. 이를 통해 특정 PyTorch 또는 CUDA 버전에 종속되지 않은 최신 버전의 DeepSpeed가 설치됩니다. DeepSpeed는 우리가 흔히 'ops'라고 부르는 여러 C++/CUDA 확장(extension)을 포함하고 있습니다. 기본적으로 이러한 모든 확장/ops는 ninja에 의존하는 torch의 JIT C++ 확장 로더를 사용하여 빌드하고 런타임에 동적으로 링크됩니다.
요구 사항 (Requirements)
- DeepSpeed 설치 전에 PyTorch가 설치되어야 합니다.
- 모든 기능을 지원하려면 PyTorch 버전이 >= 2.0인 것이 권장되며, 이상적으로는 최신 PyTorch 안정화 버전을 사용해야 합니다.
- C++/CUDA/HIP 확장을 컴파일하는 데 사용할 nvcc 또는 hipcc와 같은 CUDA 또는 ROCm 컴파일러.
- 개발 및 테스트에 사용된 특정 GPU는 아래에 나열되어 있습니다. 이는 해당 범주에 속하지 않는다고 해서 귀하의 GPU가 작동하지 않을 것이라는 의미는 아니며, 단지 DeepSpeed가 다음 환경에서 가장 잘 테스트되었음을 의미합니다:
- NVIDIA: Pascal, Volta, Ampere, 그리고 Hopper 아키텍처
- AMD: MI100 및 MI200
기여된 하드웨어 지원 (Contributed HW support)
- DeepSpeed는 이제 다양한 하드웨어 가속기(HW accelerators)를 지원합니다.
| 기여자 | 하드웨어 | 가속기 이름 | 기여자 검증 | 상위 버전 검증 |
|---|
| Huawei | Huawei Ascend NPU | npu | 예 (Yes) | 아니요 (No) |
| ... | | | | |
PyPI
저희는 PyPI에 정기적으로 릴리스를 푸시하며, 대부분의 경우 사용자들에게 이곳에서 설치할 것을 권장합니다.
설치 후에는 DeepSpeed 환경 보고서(DeepSpeed environment report)를 통해 설치 상태를 확인하고 사용자의 시스템이 어떤 확장/ops와 호환되는지 확인할 수 있습니다.
DeepSpeed 확장 기능(extensions)/연산자(ops) 중 일부를 (JIT 컴파일 대신) 미리 설치하거나, PyPI를 통해 사전 컴파일된 연산자를 설치하려면 고급 설치 지침을 참조하십시오.