Stocks such as Bank of America and EPAM Systems could benefit most from investing in AI productivity, according to Goldman Sachs.