His departure comes as the president looks at the possibility of the U.S. government acquiring stakes in AI firms