Most of everyone on here will tell you to do an AX15 swap. Here’s the problem though.
Locating a 2.5L AX15 bell housing. They are not easy to find. They came behind Dodge Dakotas between the years of 96-02 if I remember correctly. AX5 on the other hand is a fine light duty trans, you can’t beat on it as hard as you could with an AX15.