DeepSpeed

E760434

DeepSpeed is a deep learning optimization library from Microsoft that enables efficient, large-scale training of models across distributed GPU systems.

All labels observed (8)

How this entity was disambiguated

Statements (49)

Predicate Object
instanceOf deep learning optimization library ⓘ
open-source software project ⓘ
category distributed computing library ⓘ
machine learning software ⓘ
developer Microsoft ⓘ
Microsoft DeepSpeed team ⓘ
linked to: DeepSpeed
enables scaling to thousands of GPUs ⓘ
training models that do not fit in single-GPU memory ⓘ
feature 3D parallelism ⓘ
BF16 training ⓘ
DeepSpeed-Inference ⓘ
linked to: DeepSpeed

DeepSpeed-MoE ⓘ
linked to: DeepSpeed

FP16 training ⓘ
Mixture-of-Experts training support ⓘ
ZeRO optimizer ⓘ
linked to: DeepSpeed

ZeRO-Infinity ⓘ
linked to: DeepSpeed

ZeRO-Offload ⓘ
linked to: DeepSpeed

activation checkpointing ⓘ
activation partitioning ⓘ
checkpointing utilities ⓘ
communication optimization ⓘ
data parallelism ⓘ
gradient checkpointing ⓘ
mixed precision training ⓘ
model parallelism ⓘ
optimizer state partitioning ⓘ
parameter partitioning ⓘ
pipeline parallelism ⓘ
sparse attention ⓘ
tensor parallelism ⓘ
throughput benchmarking tools ⓘ
license MIT License ⓘ
optimizedFor NVIDIA GPUs ⓘ
linked to: Nvidia Maxwell GPU
programmingLanguage C++ ⓘ
Python ⓘ
repository https://github.com/microsoft/DeepSpeed ⓘ
specializesIn GPU acceleration ⓘ
distributed training ⓘ
large-scale model training ⓘ
memory optimization ⓘ
training throughput optimization ⓘ
supports model parallel training of billion-parameter models ⓘ
offloading to CPU memory ⓘ
offloading to NVMe storage ⓘ
supportsFramework PyTorch ⓘ
useCase multi-GPU training ⓘ
multi-node training ⓘ
training transformer models ⓘ
training very large language models ⓘ

How these facts were elicited

Referenced by (10)

Full triples — surface form annotated when it differs from this entity's canonical label.

NCCL → usedBy → DeepSpeed ⓘ
DeepSpeed → developer → Microsoft DeepSpeed team ⓘ
linked to: DeepSpeed
DeepSpeed → feature → ZeRO optimizer ⓘ
linked to: DeepSpeed
DeepSpeed → feature → ZeRO-Offload ⓘ
linked to: DeepSpeed
DeepSpeed → feature → ZeRO-Infinity ⓘ
linked to: DeepSpeed
DeepSpeed → feature → DeepSpeed-Inference ⓘ
linked to: DeepSpeed
DeepSpeed → feature → DeepSpeed-MoE ⓘ
linked to: DeepSpeed
Megatron-LM → influenced → Megatron-DeepSpeed ⓘ
linked to: DeepSpeed