I havent come across any official documentation but you can setup HDI with an edge node and then download and install Nifi to the edge node. Just ensure you've changed and opened the Nifi UI port. Or setup up an SSH tunnel.
Note that this ARM template is not an official MS template.
Other option is to use an existing worker node of HDI and download and install Nifi to that node.