# Updated Requirements for ICLR 2026 Submission

# Core Dependencies
torch>=2.0.0,<2.2.0
transformers>=4.35.0,<4.40.0
datasets>=2.14.0
sentence-transformers>=2.2.0
safetensors>=0.4.0
sentencepiece>=0.1.99
tokenizers>=0.15.0

# Model Serving & Inference
vllm>=0.3.0
openai>=1.0.0

# Training Infrastructure
deepspeed>=0.12.0
accelerate>=0.25.0

# Evaluation Framework
lm-eval>=0.4.0

# Math Utilities
math-verify
sympy>=1.12

# Data Processing
pandas>=2.0.0
numpy>=1.24.0,<2.0.0
jsonlines>=3.1.0
tqdm>=4.65.0

# Visualization (Optional)
matplotlib>=3.7.0
seaborn>=0.12.0

# Web Server (for reference model)
bottle>=0.12.0
tornado>=6.3.0

# Development Tools (Optional)
ipython>=8.12.0
jupyter>=1.0.0
pytest>=7.4.0

# Note: For CUDA support, ensure PyTorch is installed with appropriate CUDA version
# Example: pip install torch==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html