team lead ai ml careers in Pune, India