Sampling-based Distributed Training with Message Passing Neural Network

no code implementations • 23 Feb 2024 • Priyesh Kakka, Sheel Nidhan, Rishikesh Ranade, Jonathan F. MacArt

In this study, we introduce a domain-decomposition-based distributed training and inference approach for message-passing neural networks (MPNN).