Compute the forward pass of a 2-layer MLP with a ReLU hidden activation: h = relu(x @ W1.T + b1), output = h @ W2.T + b2.
h = relu(x @ W1.T + b1)
= h @ W2.T + b2
solve(...)