Files
llm_trainer/scripts/cal_mfu.py