I've never heard of a AW4 swap in a WJ but it would make sense. I do know that the ZJ came with a AW4 for one year.
They are bullet proof transmissions. My reverse sometimes doesn't want to cooperate but never had a problem with forward gears, 270k pampered miles.