partition function

Partition data across workers in a cluster