Write a 2-D convolution, a biased attention step and a training loop

Implement small ML building blocks from scratch: a 2-D convolution in NumPy, an attention step with an added bias term before the softmax, and the missing pieces of a standard training loop.