GNN muon energy reconstruction