Hi Nav,
The School of Data Science and AI at IIT Madras offers PhD programs in various areas. Students from diverse departments with an applied interest in AI/ML and EE/CS/AI departments are interested in fundamental research areas in AI/ML can apply. You can read more about it here.
You may also find this helpful: About Wadhwani School of Data Science and AI (WSAI)
Hope that helps!