numpy
torch
torchvision
transformers
datasets
xgboost>=2.0
scikit-learn
pillow
tensorboard

[all]
genml_kit[gcs,lora,s3,timm,uvito]

[dev]
pytest
ruff
yapf

[gcs]
google-cloud-storage

[lora]
peft>=0.7.0

[s3]
boto3>=1.28

[timm]
timm>=1.0

[uvito]
segmentation_models_pytorch>=0.3.0
