To keep every aspect of the cylinder head, you'll have to keep with the similar heads which include VVT.
so BJX will be exactly the same (
Ibiza FR/Polo GTI), AGU aswell I think.
I don't think that the AEB would be good for you as it's too old and mounted longitudinal, so little differences might not make it fit.
Stick with transverse heads.
Basically any head from 2002-2008 should fit perfectly.