Yes and no.
The AX-15 is the same transmission, but the bellhousing will be different. The Dakota (the AX-15 will be behind the 238ci V6) will have an LA-Block bellhousing pattern, while the Jeep version will have the AMC "Modern Era I6" and Gen-II/III V8 pattern.
You can piece the upgrade together if you have to. I believe (don't hold me to it!) that the transfer case is clocked the same amount in the XJ, MJ, ZJ, and Dakota; while it's clocked flatter in the TJ and YJ. (Yes, the AX-15 was used in the ZJ - 1993-1993.5 w/242ci I6 only. It's more rare than an AW4-equipped ZJ - 1993-1994.5 w/242ci, then it was replaced with a modified 42RE.)
If you get your source unit from a TJ or YJ; either the transmission tunnel must be modified to clear the transfer case, or the transfer case adapter housing must be modified to "reclock" the transfer case (or stick in an adapter ring.)
These are issues if you're converting from the BA-10/5 to the AX-15. If you're replacing an AX-15 that's failed, then you can reuse the bellhousing and associated parts up front, the transfer case adapter housing from the rear, and simply swap out the transmission proper (you'll still need to have the same version of the AX-15 - 1988.5-1993 [Jeep apps] had the "internal" slave cylinder [I refer to that version as the "AX-15i",] and the 1994-up had the "external" slave cylinder [referred to as "AX-15e."])
NB: The NV3550 uses the same bellhousing and associated clutch parts (save the front bearing retainer/sleeve assembly) as the AX-15e...
"recon" (sic - reckon)(tm) "hihgly"(tm) "seceed"(tm)
"Outback AIDS - Alcohol-Induced Dizzy Spells"