# Core dependencies
torch>=1.12.0
torchvision>=0.13.0
numpy>=1.21.0
rich>=12.0.0
click>=8.0.0  # For CLI interface
pytz>=2021.3  # For timezone handling in experiment directories

# Optional dependencies for enhanced functionality
torchinfo>=1.7.0  # For model summary
scikit-learn>=1.0.0  # For additional metrics
psutil>=5.8.0  # For hardware monitoring (CPU, RAM usage)
pynvml>=8.0.4  # For GPU monitoring (NVIDIA GPUs) 