The head of the United Kingdom's nuclear submarine capability has underlined WA's critical role in driving the success of the controversial AUKUS program during a visit to Canberra.