torch==2.8.0
triton==3.4.0
numpy
flash_attn
einops
pandas
matplotlib
pickle
tqdm
