openteamsinc
pypi/deepspeed
Risk Profile
Package
Source
Risk Profile
Maturity:
Mature
Health:
Healthy
Legal:
Moderate Risk
Package license does not match the source code license
License includes a patent grant clause
Security:
Caution Needed
At least one high or critical severity vulnerability has been reported in the last 600
Package
pypi
deepspeed
Version
0.17.5
Last Release Date
19 days ago
License
Apache Software License 2.0
Dependencies
einops
hjson
msgpack
ninja
numpy
packaging
>=20.0
psutil
py-cpuinfo
pydantic
>=2.0.0
torch
tqdm
1bit-mpi:
mpi4py
readthedocs:
autodoc_pydantic
>=2.0.0
docutils
<0.18
hjson
packaging
psutil
py-cpuinfo
pydantic
>=2.0.0
recommonmark
sphinx_rtd_theme
torch
tqdm
dev:
accelerate
clang-format
==18.1.3
comet_ml
>=3.41.0
deepspeed-kernels
docutils
<0.18
future
importlib-metadata
>=4
mup
pre-commit
>=3.2.0
pytest
<8.4.0, >=7.2.0
pytest-forked
pytest-randomly
pytest-xdist
qtorch
==0.3.0
recommonmark
sphinx
sphinx-rtd-theme
tensorboard
torchvision
transformers
>=4.51.3
wandb
autotuning:
tabulate
autotuning-ml:
hjson
tabulate
xgboost
sparse-attn:
triton
==1.0.0
sparse:
neural-compressor
==2.1.0
inf:
google
lm-eval
==0.3.0
protobuf
qtorch
safetensors
sentencepiece
transformers
>=4.32.1
sd:
diffusers
>=0.25.0
triton
>=2.1.0
triton:
triton
==2.1.0
deepcompile:
scipy
all:
qtorch
==0.3.0
sphinx
clang-format
==18.1.3
comet_ml
>=3.41.0
pytest-randomly
packaging
tensorboard
tqdm
triton
==2.1.0
pytest-xdist
deepspeed-kernels
sphinx-rtd-theme
psutil
pytest
<8.4.0, >=7.2.0
wandb
pytest-forked
accelerate
sentencepiece
qtorch
sphinx_rtd_theme
protobuf
xgboost
triton
>=2.1.0
lm-eval
==0.3.0
scipy
safetensors
neural-compressor
==2.1.0
hjson
autodoc_pydantic
>=2.0.0
pre-commit
>=3.2.0
google
py-cpuinfo
importlib-metadata
>=4
docutils
<0.18
transformers
>=4.51.3
tabulate
pydantic
>=2.0.0
mpi4py
recommonmark
mup
torch
torchvision
diffusers
>=0.25.0
transformers
>=4.32.1
future
triton
==1.0.0
Source
Location
deepspeedai/DeepSpeed
Last Source Update
3 days ago
Licenses
Apache License 2.0
Apache License 2.0
Distribution Destinations
pypi/deepspeed
