googlehydrology.modelzoo.mean_embedding_forecast_lstm module