if ur AX3000 is global rom then yes, can replace the TP link.
im using global version AX3200 which support vlan tagging

there is 2 vlan tagging for unifi, i also not very familiar but can google to get answer. if not mistaken is 500 for internet n 600 for iptv.
in my case i din use the unifi TV so not a concern to me to set up it