torch>=2.0
numpy>=1.24
matplotlib>=3.7
# real-token stage only
datasets>=2.14
transformers>=4.30
