import numpy as np def mlp_forward(x, params): raise NotImplementedError def mse_loss(predicted, target): raise NotImplementedError def mlp_backward(params, cache, target): raise NotImplementedError