As far as I can tell they used the same trans A340F from 96-2002 but when I do a search 2001 trans is twice as much dealing with the milkshake right now
If you look on Craig's List the parts houses that part out a runner always group the 96-98 together when listing what they fit. A AW4 out of a jeep is also the same trans. Different bellhousing.