One can swap any Toyota 4cylinder engine in ae chassis.
4age is good option but if you ask me as I own 86 and going to swap soon for drag+daily. i'm preferring 4agze or 2zzge.2zzge is new engine having new technology awesome performance and very well upgrade strength but slightly expensive...was getting in 150k LSD while 4agze is old and rare...was really hard to get in good condition specially original whereas I'm getting it for 80k.
Both engines are best of all.I prefer them more then 3sgte because 3sgte weights to much for ae80.
One more thing 2zzge and 4agze are not easy to handle on lightweight ae80.