ML, Data Science role landscape in Mumbai/Pune

I'm currently based out of BLR and due to personal reasons I might be making a switch to move to Mumbai or Pune. I was wondering what good orgs are located in Mumbai-Pune that can offer AI/ML related roles (excluding Banks and Financial Institutions, Big4) Or am I better off with Bengaluru?