See this Medium post: Graph Transformer for Node Label Prediction with PyG