step,epoch,train_loss,val_loss,learning_rate,grad_norm,step_time_ms