The US and UK are teaming up to test the safety of AI models