Embedding AI in critical infrastructure partnerships