sorry, I didn’t use the normalized X into dataset.  With normalized ds, here is the training/validation loss after 2 epochs:

training loss: 0.2389
validation loss: 0.2239

here is the outputs of the first training data batch:

tensor([[2.2860],
[2.0087],
[2.1482],
[1.7892]], grad_fn=)

what concerns me now is that the the loss increases by increasing the epochs.  For example here are the loss for just one epoch:

training loss: 0.1174
validation loss: 0.1241

is this normal behavior?