Minjia Zhang, Microsoft AI and Research