numpy
PyYAML
h5py
tqdm
wget
paho-mqtt
boto3
pynvml
sklearn
networkx
click
torch
torchvision
spacy
gensim
multiprocess
smart-open==5.2.1
nvidia-ml-py3
matplotlib
dill
pandas
wandb==0.13.2
eciespy
MNN==1.1.6

[MPI]
mpi4py

[gRPC]
grpcio

[jax]
jax[cpu]
dm-haiku
optax
jaxlib

[mxnet]
mxnet==2.0.0b1

[tensorflow]
tensorflow
tensorflow_datasets
tensorflow_federated
