US proposes AI safety notification mechanism in talks with China