ML-based Reconstruction of the SND@LHC detector GNN training setup, implemented with pytorch Inputs created with https://github.com/steggema/snd-scripts Setup inspired by https://github.com/murnanedaniel/GeometricAttention