SOURCES EMBEDDINGS CONTEXT PREDICTION P(y) h_t WEATHER FORCING SATELLITE (N/A) GAUGE HISTORY EMBEDDING W EMBEDDING S EMBEDDING G MASKED MEAN BASIN CONTEXT LSTM STACK PROB OUTPUT