PyPI page
Home page
Author:
DeepSpeed Team
License:
Apache Software License 2.0
Summary:
DeepSpeed library
Latest version:
0.16.2
Required dependencies:
deepspeed-kernels
|
einops
|
hjson
|
msgpack
|
ninja
|
numpy
|
packaging
|
psutil
|
py-cpuinfo
|
pydantic
|
torch
|
tqdm
Optional dependencies:
accelerate
|
autodoc_pydantic
|
clang-format
|
comet_ml
|
diffusers
|
docutils
|
future
|
google
|
hjson
|
importlib-metadata
|
lm-eval
|
mpi4py
|
mup
|
neural-compressor
|
packaging
|
pre-commit
|
protobuf
|
psutil
|
py-cpuinfo
|
pydantic
|
pytest
|
pytest-forked
|
pytest-randomly
|
pytest-xdist
|
qtorch
|
recommonmark
|
safetensors
|
sentencepiece
|
sphinx
|
sphinx-rtd-theme
|
sphinx_rtd_theme
|
tabulate
|
tensorboard
|
torch
|
torchvision
|
tqdm
|
transformers
|
triton
|
wandb
|
xgboost
Downloads last day:
18,244
Downloads last week:
153,812
Downloads last month:
577,420