High-level parallelization in Python