Elon Musk’s xAI is now working with the federal government