The Xbox SSDs are all PCIe 4.0x2 so you need WD CH (NOT PC) SN530/SN560 or SSSDTC/Lite-On XA1 SSDs.
The CH SN560 is a newer model that has faster read speeds (2400 vs 2000 MB/s) but is much more rare. Most of the CH WD SSDs on the market is CH SN530.