The R154 is close to the AX15 and shares lots of parts, but note that while the R154 was indeed used behind the Supra, those that built up their 2ZJ's often found the manual trans to be the weak spot in the drivetrain and most running seriously boosted engines all went with an automatic (or in some instances modified a Tremec to fit).
Personally, I don't think I'd ever consider an AX15 a viable candidate in stock form. RSG does makes an upgraded "Terminator" AX15 that uses some upgraded gears and internals of the later Supra R154 transmissions. Supposedly, it is typically quite reliable behind LS-powered Jeeps. The NV4500 is another option, though in my opinion the NV is a big ol' thing and since it has been out of production since the mid-2000's, it is getting harder to find. It's often too long for non-stretched Jeeps too!
The one option that I'd be most inclined to try out is the new(ish) Tremec TR-4050.
https://tremec-blog.com/tremec-tr-4050-off-road-manual-transmission-solution/
It has an excellent set of gear ratios: 6.16, 3.11, 1.71, 1.00, 0.76 which would work well on the street and trail. That 6.16 first gear with a 2.72 t-case Lo range, and 5.13 axle ratio would yield a ~86:1 crawl ratio - that is higher than an 5-speed TJ Rubi. If you were to couple the SST with a Rubicon t-case, you would end up with a 126:1 crawl ratio - enough to make even the double/triple t-cased Toyota guys jealous!