# Core ML dependencies
torch>=2.2.0
transformers>=4.39.0
accelerate>=1.10.1
huggingface-hub>=0.34.4

# Configuration and data handling
pydantic>=2.5.0
PyYAML>=6.0

# Logging and experiment tracking
wandb>=0.21.3

# Metric learning for diversity loss
pytorch-metric-learning>=2.9.0

# Utilities
pandas>=2.3.2
numpy>=1.24.0
