Representing sub-grid heterogeneity of cloud and precipitation across scales