Vectorized NumPy Model Implementation

misprinting
dimension should be [1,j_out] not [m,j_out]

Hello @iitk_gaurav ,

Thank you very much. WIll share this with the team.

Raymond