Learning composable energy surrogates Code for the paper "Learning Composable Energy Surrogates for PDE Order Reduction", NeurIPS 2020.