ENH: use scientific notation for progress logging

[Xylambda]

  msg = (
      f"Epoch {epoch_number}/{total_epochs} | Train loss: "
      f"{train_loss:.2e} | Validation loss {val_loss:.2e} | "
      f"Time/epoch: {epoch_time:.2f} s"
  )`