ML Coding with NP and NN Layers

Solve a ML-based puzzle and implement it in code. Familiarity with vector and matrix addition/multiplication in numpy, common neural network layers, and implementation of linear layers with batched input in numpy is useful. Example Input 2 1 1 1 1