src/transformers/models/graphormer/modeling_graphormer.py/GraphormerPreTrainedModel/normal_
class GraphormerPreTrainedModel:    def normal_(self, data):
        # with FSDP, module params will be on CUDA, so we cast them back to CPU
        # so that the RNG is consistent with and without FSDP
        data.copy_(data.cpu().normal_(mean=0.0, std=0.02).to(data.device))