Penny Wong to discuss Australia’s plans for nuclear submarines with International Atomic Energy Agency chief