Matrix multiplication is the core operation behind linear layers, attention, and many ML
Write a function matrix_multiply(A, B) that returns the matrix product of A and B.
The inputs can be Python lists or NumPy/Torch tensors, but the output should be a 2D array of
numbers representing the product.