dill
pandas
psutil
requests
rich
termcolor
tqdm

[all]
tracebloc[lifelines,pytorch,scikit-survival,sklearn,tensorflow]

[catboost]
catboost>=1.2.10

[dev]
pytest>=7.0
pytest-cov>=4.0
mypy>=1.5
ruff>=0.1

[lifelines]
lifelines>=0.30.3

[lightgbm]
lightgbm>=4.6.0

[pytorch]
numpy
Pillow
torch>=2.11.0
torchvision>=0.26.0
torchmetrics>=1.9.0
timm>=1.0.26
pycox>=0.3.0
pytorch-forecasting>=1.4.0
transformers>=5.8.0
peft>=0.19.1
torchao>=0.17.0

[scikit-survival]
scikit-survival>=0.27.0

[sklearn]
numpy
scikit-learn>=1.8.0
interpret>=0.6.9
catboost>=1.2.10
lightgbm>=4.6.0
xgboost>=3.2.0

[tensorflow]
tensorflow<2.21,>=2.19
tensorflow-datasets>=4.8
silence-tensorflow>=1.2

[xgboost]
xgboost>=3.2.0
