Model Parallelism Numpy Documentation Example