Rollout assignment balance control #535
Description
From rollout results assigning to policy replica, according to their size to make all replicas as balanced as possible.
Also consider the balance among all ranks and balance among items in one batch, all inside one replica.