Google is dealing with an AI talent shortage — and wants the U.S. to change immigration policy to fix it