numpy
multiprocess
scikit-learn
cardinality
sultan
tqdm
scipy
gdown
tensorly

[develop]
pytest
pytest-asyncio
pytest-cov
pytest-xdist
coverage
jinja2
tensorflow<2.11
tensorflow_datasets
tensorflow_hub
torch
torchvision
torchtext
torchdata
portalocker
datasets
grpcio
protobuf
grpcio-tools

[distributed]
grpcio
protobuf

[hugginface]
datasets

[pytorch]
torch
torchvision
torchtext
torchdata
portalocker

[tensorflow]
tensorflow<2.11
tensorflow_datasets
tensorflow_hub
