Like AoA said before I use the ADATA - XPG GAMMIX S70 Blade 2TB Internal SSD PCIe Gen 4 x4 but I got the Sabrent M.2 NVMe PS5 heatsink and it has worked great.
Paid $250 a year ago and it is down to $180 on Amazon right now. And $20 for the heatsink.