How can we create better climate models? We tackle this by proposing a data-driven successor to the existing approach for capturing key temporal trends in climate models. We combine probability, allowing us to represent uncertainty, with machine learning, a technique to learn relationships from data which are undiscoverable to humans. Our model is often superior to existing baselines when tested in a simple atmospheric simulation.