I have experience in swapping out a BUNCH of pickups in Axis and Axis SS models, and for your situation I would go with the Dimarzio AT-1 in the bridge and and PAF 36th Anniversary in the neck (DiMarzio also recommends pairing the 36th Anniversary with the AT-1).
The DiMarzio Tone Zone and Super Distortion in the bridge would also work, but I think the AT-1 is the way to go. Zebra is always my favorite.
I currently have a set of BNP Polymaths in my Axis, and they are fabulous and less compressed than stock, but I did have to add a tone circuit in the control cavity. As a side note, EVH also hid tone circuits on occasion, with the most notable being on the Bumblebee guitar (at least that we know of) for some of the tracks on WACF. If you have never checked out Jim Gaustad’s YouTube page and his deep dives on EVH tone, please take a few minutes - it is a fabulous resource.