# Core dependencies
torch>=1.12.0
torchvision>=0.13.0
numpy>=1.21.0
rich>=12.0.0

# Optional dependencies for enhanced functionality
torchinfo>=1.7.0  # For model summary
scikit-learn>=1.0.0  # For additional metrics 