I presented our work at the 'Data-Driven Physical Simulations' seminar series at the Lawrence Livermore National Laboratory. Thank you very much to Youngsoo Choi from the Data Science Institute at LLNL for organizing the seminar series and for inviting me!