JVP and VJP for a 2-Layer Network | TensorTonic