You'll need to be careful to match the bellhousing pattern (the part that bolts to the engine). While AX-15s were used in other vehicles, the bellhousing may differ, requiring an adapter or making it incompatible. Sticking with an AX-15 from a similar-year Jeep Wrangler will guarantee compatibility.
Before purchasing a used transmission, verify the transmission's condition. A used transmission in poor condition can lead to more problems than it solves.