AI, Elon Musk's AI company, has partnered with the US government to expand access to the Grok AI chatbot for federal use.