Computes the mean squared error
See also
Other luz_metrics:
luz_metric(),
luz_metric_accuracy(),
luz_metric_binary_accuracy(),
luz_metric_binary_accuracy_with_logits(),
luz_metric_binary_auroc(),
luz_metric_mae(),
luz_metric_multiclass_auroc(),
luz_metric_rmse()