openteamsinc
pypi/deepspeed
Risk Profile
Package
Source
Risk Profile
Maturity:
Mature
Health:
Healthy
Legal:
Moderate Risk
License includes a patent grant clause
Package license does not match the source code license
Security:
Healthy
Package
pypi
deepspeed
Version
0.19.6
Last Release Date
about 16 hours ago
License
Apache Software License 2.0
Dependencies
einops
hjson
msgpack
ninja
numpy
packaging
>=20.0
psutil
py-cpuinfo
pydantic
>=2.0.0
torch
>=2.0.0
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
>=2.0.0
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
torchembed:
torchembed
>=0.3.1
all:
importlib-metadata
>=4
pytest-xdist
psutil
torch
>=2.0.0
tqdm
sentencepiece
triton
==1.0.0
torchembed
>=0.3.1
lm-eval
==0.3.0
pytest
<8.4.0, >=7.2.0
pre-commit
>=3.2.0
sphinx-rtd-theme
diffusers
>=0.25.0
wandb
docutils
<0.18
packaging
hjson
tabulate
google
future
pydantic
>=2.0.0
neural-compressor
==2.1.0
qtorch
scipy
comet_ml
>=3.41.0
triton
==2.1.0
protobuf
accelerate
autodoc_pydantic
>=2.0.0
recommonmark
transformers
>=4.32.1
clang-format
==18.1.3
pytest-forked
deepspeed-kernels
xgboost
safetensors
sphinx_rtd_theme
qtorch
==0.3.0
mpi4py
mup
transformers
>=4.51.3
py-cpuinfo
tensorboard
sphinx
triton
>=2.1.0
pytest-randomly
torchvision
Source
Location
deepspeedai/DeepSpeed
Last Source Update
about 18 hours ago
Licenses
Apache License 2.0
Apache License 2.0
Distribution Destinations
pypi/deepspeed
