torch>=2.0.0
torchvision>=0.15.0
accelerate>=0.20.0
numpy>=1.24.0
scipy>=1.10.0
scikit-learn>=1.2.0
pandas>=2.0.0
matplotlib>=3.7.0
tqdm>=4.65.0
wandb>=0.15.0
pyyaml>=6.0.0
h5py>=3.8.0
safetensors>=0.3.0

[all]
pytest>=7.0.0
pytest-xdist>=3.5.0
ruff>=0.8.0
pre-commit>=3.5.0
onnx>=1.14.0
onnxruntime>=1.15.0
triton
optuna>=3.0.0

[compile]
triton

[dev]
pytest>=7.0.0
pytest-xdist>=3.5.0
ruff>=0.8.0
pre-commit>=3.5.0

[hpo]
optuna>=3.0.0

[onnx]
onnx>=1.14.0
onnxruntime>=1.15.0
