US says New Zealand welcome to "engage" in AUKUS