Change compute_loss name to loss_balancing + modification of the...
Change compute_loss name to loss_balancing + modification of the MultiLossBalancing class to remove the loss computation out of the forward function + modification of the train/eval_step_mt according to this new loss computing technique