Perfect mate up would be an Audi A3, since it shares the transverse engine layout, and is essentially a platform twin to the A4-generation (don't confuse the designations) G/J/NB.
Installing it into any other Audi requires more work but is definitely possible but with more effort and cross-swapping of parts.