White House AI policy adviser Sriram Krishnan will leave at the end of June, days after the administration floated taking government equity stakes in AI companies.