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.2
Last Release Date
10 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:
sphinx
tabulate
importlib-metadata
>=4
transformers
>=4.51.3
triton
==2.1.0
hjson
pytest
<8.4.0, >=7.2.0
recommonmark
pytest-forked
wandb
pytest-randomly
neural-compressor
==2.1.0
deepspeed-kernels
accelerate
scipy
mup
sentencepiece
transformers
>=4.32.1
google
lm-eval
==0.3.0
sphinx_rtd_theme
future
pydantic
>=2.0.0
xgboost
packaging
torchvision
torch
tensorboard
pytest-xdist
qtorch
safetensors
clang-format
==18.1.3
pre-commit
>=3.2.0
qtorch
==0.3.0
py-cpuinfo
triton
>=2.1.0
sphinx-rtd-theme
autodoc_pydantic
>=2.0.0
psutil
tqdm
triton
==1.0.0
mpi4py
protobuf
diffusers
>=0.25.0
comet_ml
>=3.41.0
docutils
<0.18
Source
Location
deepspeedai/DeepSpeed
Last Source Update
3 days ago
Licenses
Apache License 2.0
Apache License 2.0
Distribution Destinations
pypi/deepspeed
