Files
ai-toolkit/scripts/calculate_timestep_weighing_flex.py