requirements.txt

# Core dependencies for AugMP federated fine-tuning simulation with GPU support
torch>=2.0.0
transformers>=4.37.0,<5.0.0  # 4.37+ for Qwen2/Qwen2.5 support
numpy>=1.21.0
pandas>=1.3.0
scipy>=1.7.0  # Required for ALIE attack (scipy.stats.norm)
tqdm>=4.62.0
matplotlib>=3.4.0
seaborn>=0.12.0  # Required for seaborn-v0_8-darkgrid style support
peft>=0.6.0  # For LoRA support

datasets>=2.0.0  # Required for IMDB, DBpedia, Yahoo Answers datasets

# Optional: Machine learning utilities (for future use, currently unused)
# scikit-learn>=1.0.0

# Optional: Experiment tracking (currently unused)
tensorboard>=2.10.0

# Optional: TPU support (install only if using TPU)
# torch-xla==2.4.0  # Uncomment if using TPU
# cloud-tpu-client==0.10  # Uncomment if using TPU

# Optional: Accelerated training
# accelerate>=0.20.0  # For distributed training support