Introduction to model parallelism and multi GPU training