# Core training stack
torch>=2.2
torchvision>=0.17
numpy>=1.24
Pillow>=10.0
PyYAML>=6.0

# Medical segmentation losses and metrics
monai>=1.3

# XNet2d wavelet and VMamba dependencies
ptwt>=0.1.9
PyWavelets>=1.5
timm>=1.0
fvcore>=0.1.5
einops>=0.7
packaging>=23.0
triton>=2.2; platform_system == "Linux"

# Experiment logging and utilities
swanlab>=0.6
tqdm>=4.66
matplotlib>=3.8

# Tests
pytest>=8.0