numpy>=1.21
rich>=13.0
psutil>=5.9

[all]
torch>=2.2
nvidia-ml-py>=12.535.133
wandb>=0.15
mlflow>=2.0
transformers>=4.30
accelerate>=0.20
lightning>=2.0

[gpu]
nvidia-ml-py>=12.535.133

[hf]
transformers>=4.30
accelerate>=0.20

[lightning]
lightning>=2.0

[torch]
torch>=2.2

[trackers]
wandb>=0.15
mlflow>=2.0
