AI is expected to remain a key growth driver for the stock market this year.