Add GNN training

• Add GNN training with full graph
• Add GNN training with local graph (~ PF region)