See there are lateral entry examination through which you can take admission to BTech different course after Diploma. So technically, yeah! its possible for you to take admission in the course of CSE(AI & ML). Now there are not much colleges in India which offers the course CSE (AI & ML ) lateral entry course, so in general you have to take admission to the course BTech CSE lateral entry as most of the colleges offers this course in BTech.

Now if you wish to know the college, then I would recommend you to repost the question with the name of the state or place where you are finding the college.

For more information regarding BTech Lateral Entry consider the link below:

https://engineering.careers360.com/articles/what-btech-lateral-entry-eligibility

I hope this helps. Wish you a great day!