US President Donald Trump has flagged his support for AUKUS after the Australian government paid the US $798m as part of the deal.