Training speedups via batching for geometric learning: an analysis of static and dynamic algorithms

Daniel T. Speckhard, Tim Bechtel, Sebastian Kehl +2

Graph neural networks (GNN) have shown promising results for several domains such as materials science, chemistry, and the social sciences. GNN models often contain millions of par…