US treasury chief says Washington will work with Beijing on AI safety protocols while ensuring innovation and technological leadership remain unaffected.