We demonstrate a machine learning method to estimate the temperature changes in the Earth's crust over time. The method respects physical laws and conditions imposed by users. By using observed rock cooling ages as constraints, the method can be used to estimate the tectonic and landscape evolution of the Earth. We show the applications of the method using a synthetic rock uplift model in 1D and an evolution model of a real mountain range in 3D.