Implement the Adam optimizer to train a linear regression model using mean squared error (MSE) loss with full-batch gradient descent.
