feat: pluggable "work" for partitions

[zenyui]

Makes an interface for computing work/load in a partition.

Currently, rebalancing is only based on number of partitions per node, but we should make that customizable

Current implementation can be the default work implementation