Model Parallelism Numpy Challenges