Implement a learning rate sweep that trains a linear model on the given data with multiple learning rates and returns the loss curve for each.
