Optimism over the profit potential of artificial intelligence has helped drive the U.S. stock market to record highs, but investors are looking for weak spots that could emerge in the AI trade and have identified some risks to watch for.