Is "Hierarchical All-to-all" feat available in current version?

According to https://arxiv.org/pdf/2201.05596, we are very interested on "Hierarchical All-to-all" and think this is an amazing feature. But after reading source code, we could not find any part related to this. I found that team said that they did not have a ETA for this  in https://github.com/microsoft/DeepSpeed/issues/2039, will this be supported in the future?