AI-enabled production forecasting