From my understanding, most of the VW VE pumps, save for some of the later AAZ pumps, will have a 17mm mainshaft, which some say is potentially weak for the larger head/plunger needed to fuel the TDI.
If you don't need insane amounts of fuel, that'll be fine, not sure what your goals are, if you need more power/fuel, then the cummins pump is the common next step up.
personally, I plan on the AAZ pump (unsure if I have a 17mm or the 'possible' 20mm) with a 10mm or 11mm head with either a modified AAZ boost pin, or a 1.6TD pin. As well as a lift pump of some sort.
but this is just what I've read, I haven't done anything yet specifically, someone else can confirm?