The AW-4 is the only trans ever offered behind the 4.0 in a Cherokee. The early Wrangler had a Chrysler A-999 three-speed that would be a mechanical bolt in, and I think there were a couple different Autos offered in the Grand Cheerokee with a six. These would all "bolt in" without a ton of adapters, etc.
Problem is:
A-999: No overdrive, purely hydraulic, Engine computer would probably have a fit.
Anything out of a Grand: Transmission probably wouldn't like the control signals from the engine computer, and the engine computer would, again probably have a fit. Any swap would probably require a new engine computer, and a separate controller for the trans.
Any thing else is fab, Fab, FAB!
I'd say get a spare AW-4 if your worried about it breaking.