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