torch==2.6.0
numpy==1.26.4
matplotlib==3.10.3
seaborn==0.13.2
openai==1.108.0
pydantic==2.11.9
tqdm==4.67.1
wandb==0.18.7
argparse
