plan partitioning for mpi
Fix: no longer assume the index of the first element of a partition is a multiple of the partition size
Adapt coloring for MPI
(OpenMP) remove unused args from generated code
(OpenMP) push color iteration up to python
(OpenMP) add MPI support