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