AI in the Cosmos