Refactor and speed up cirq.transformers.stratify
#6013
Merged
cirq.transformers.stratify
#6013