torch-tb-profiler<=0.4.3
scikit-learn<=1.6.1
scipy<=1.16.1
numpy<=2.2.1
onnx<=1.17.0
onnxscript<=0.2.0
onnxruntime<=1.20.0
optuna<=4.1.1
mlflow<=2.19.1
kornia<=0.7.4
albumentations<=1.4.25
pytest<=8.3.5
mypy<=1.15.0
seaborn<=0.13.3
matplotlib<=3.10.0
colorama<=0.4.6
msgpack<=1.1.0
torchinfo<=1.8.0

[:platform_machine == "x86_64"]
torch<=2.6.0
torchvision<=0.21.0
lietorch<=0.6.2.post1
pynvml
pycuda
torchvision
norse
tonic
aestream
expelliarmus
