I might be swapping in a 258 or stroked 4.0 into my 715 if my previous power plant options aren't going to work out. The issue with the aw4 is heat, it creates a ton of it. With a large transmission cooler I have seen it hold up to some serious abuse. The AX15 on the other hand was initially designed for 1/2 ton drive trains like the aw4, but it's gearing is not adapt for a 5,000+ LB vehicle where the aw4 could take the excess weight displacement with the proper modifications. Also the AX15 is easily confused with the AX5 as the two share many similarities. Tall axle gearing would help but unless swapping in newer one tons the 5:89's may not be enough. Toss in a NV4500 for its lower overall ratios, mate it to a twin-sticked lomax 4:1 equipped dana 300 and keep it jeep!
(just a very broad thought)
RM
If it ain't broke, it will be.