Who is better suited for an AI product management role compared to a traditional pm role?

This is not a simple question. There is no such thing as a "traditional PM" because PM roles and responsibilities differ by industry segment. There are substantial differences in PM roles from consumer app to enterprise app to hardware and enterprise infrastructure. 

Fundamentally the role of the PM is still the same - to consider customer problems, potential solution approaches, and lead them from concept to launch and facilitate a continous innovation lifecycle. The only difference here in AI/ML PM is a new approch to solving the problem with AI/ML tools and frameworks. So anyone with a "Learning mindset", ability to pick up on the AI/ML technologies and comfortable in ambiguity as this field is still growing, -- would do very well. 

5 years from now, likely there is going to be no difference between a Traditional PM and AI PM. AI is going to be used/present in all products. I see "Traditional PM role" as a foundational one to have, upon which one can grow to become a good AI PM. Good Traditional PM with aptitude for tech and data science is likely to do well as a good AI PM. Taking a Udemy course on basics of AI/ML, and applying to every day PM job will be a great start.