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?