# RL-Driven Model Selection Training Dependencies
# Install with: pip install -r requirements.txt

# Core ML/DL
torch>=2.0.0
transformers>=4.35.0
accelerate>=0.24.0
peft>=0.6.0
bitsandbytes>=0.41.0

# RL Training
trl>=0.7.0  # Transformer Reinforcement Learning
stable-baselines3>=2.0.0

# Graph Neural Networks (for GMTRouter)
torch-geometric>=2.4.0
torch-sparse>=0.6.17
torch-scatter>=2.1.2

# Data Processing
datasets>=2.14.0
pandas>=2.0.0
numpy>=1.24.0
scipy>=1.11.0

# Evaluation
scikit-learn>=1.3.0
rouge-score>=0.1.2

# Utilities
pyyaml>=6.0
tqdm>=4.65.0
wandb>=0.15.0  # Optional: experiment tracking
tensorboard>=2.14.0

# HTTP/API for data collection
requests>=2.31.0
aiohttp>=3.8.0
