def build_model_gbuf_range_map(cls, model):
        """
        Create param-to-grad-buffer mappings, for grad buffer data types
        within a specific virtual model.
        """
        return {
            dtype : cls.build_model_gbuf_range(model, dtype)
            for dtype in model._grad_buffers
        }

        Create param-to-grad-buffer mappings, for grad buffer data types
        within a specific virtual model.
        
training/DeepSpeed-Domino/domino/optimizer/distrib_optimizer.py