Based on: Kharazmi, Ehsan, Zhongqiang Zhang, and George E. Karniadakis. "hp-VPINNs: Variational Physics-Informed Neural Networks With Domain Decomposition." arXiv preprint arXiv:2003.05385 (2020).