AtomFlow: Accelerating GNN-Based Molecular Dynamics with ab initio Accuracy on Fugaku Supercomputer
This paper proposes AtomFlow, a holistic acceleration for a representative GNN-based MD model on the Fugaku supercomputer, and redesigns the inference workflow with a framework-free implementation to avoid the overhead of the PyTorch framework on Fugaku.