I would go for .ai because thats the core feature. domain with .ai gives away your core offering (i.e. ML/Al) and you would have customer attention from the very beginning. According to me, it also adds credibility and legitimacy in a subtle way which goes a long way in lead conversion.