pyyaml>=6.0
numpy>=2.0
pandas>=2.2
scikit-learn>=1.5
torch>=2.4
torchvision
torchaudio
tensorboard>=2.18
lightning>=2.4
matplotlib>=3.9
seaborn>=0.13
clustergram>=0.8
pytest>=7.0

[gpu]
cudf-cu12>=24.10
cuml-cu12>=24.10
