GPUtil
PyYAML
aiohttp>=3.8.1
attrdict
attrs
boto3
cachetools
chardet
click
dill
docker==6.1.3
fastapi
gensim
geventhttpclient<=2.0.9,>=1.4.4
graphviz<0.9.0,>=0.8.1
h5py
httpx
matplotlib
multiprocess
networkx<3.0
ntplib
numpy<2.0.0
onnx
paho-mqtt<2.0.0
pandas
prettytable
py-machineid
pydantic
pytest
pytest-mock
python-rapidjson>=0.9.1
redis
scikit-learn
smart-open==6.3.0
sqlalchemy
toposort
torch>=1.13.1
torchvision>=0.14.1
tqdm
tritonclient
typing_extensions
tzlocal
uvicorn
wandb==0.13.2
wget
requests<2.32
python-dotenv
protobuf<4.0dev,>=3.20.2
typer<0.10.0,>=0.3.0
fastapi-cli==0.0.1

[MPI]
mpi4py

[deepspeed]
deepspeed>=0.10.2

[fhe]
tenseal

[gRPC]
grpcio

[jax]

[llm]
accelerate>=0.24.0
datasets>=2.14.0
einops
evaluate
ninja
packaging
peft>=0.4.0
safetensors
sentencepiece
transformers[torch]>=4.31.0
zstandard

[mxnet]
mxnet==2.0.0b1

[nlp]
spacy<3.3.0,>=3.2.0

[tensorflow]
tensorflow
tensorflow_datasets
tensorflow_federated
