Can anyone help identify transmission?

When looking up A999 transmission i get a majority search of jeep transmission. Obviously this is for a rear wheel drive and not 4wd. When looking for rebuild, searches come from torqueflight 6, A908/A998/A999 rebuild kit. Considering the A999 is a 5 disc as opposed to its predecessors, can you give me a suggestion as for what rebuild kit or if there’s an upgrade valve body for the A999? Would like to keep the lock up. I’m assuming it’s used as an external lock up with toggle switch on the dash?

I've used these guys in the past:
A904 Rebuild Kit A998 A999 Automatic Transmission Master Overhaul Banner Box Set Torqueflite 6
I'm sure there are a little better deals out there if you're willing to do more searching.