'I'll be working on helping tackle some of the large challenges facing America on AI (more on that later),' Krishnan wrote in a post on X