Distributed Machine Learning for Computational Engineering using MPI

Kailai Xu, Weiqiang Zhu, Eric Darve

We propose a framework for training neural networks that are coupled with partial differential equations (PDEs) in a parallel computing environment. Unlike most distributed computi…