docker
google-cloud-storage
grpcio
matplotlib
nltk
numpy
pandas
sly
tqdm
wandb

[dev]
black
flake8
flake8-quotes
grpcio-tools
isort
mypy
mypy-protobuf
pre-commit
pytest
rbql
rstcheck
sphinx==4.0.2

[full]
ml2[hf,jp,pt,ray,tf]

[hf]
datasets
transformers

[jp]
jupyter

[package]
setuptools>=46.4
wheel
build
twine

[pt]
torch

[ray]
ray[default,tune]

[tf]
tensorflow>=2.13

[tf-legacy]

[tf-legacy:platform_machine != "arm64" or platform_system != "Darwin"]
tensorflow<2.12,>=2.10.0

[tf-legacy:platform_machine == "arm64" and platform_system == "Darwin"]
tensorflow-macos<2.12,>=2.10.0

[tf:platform_machine == "arm64" and platform_system == "Darwin"]
tensorflow-metal>=1.0
