First I tried an aftermarket ek low profile heatsink. It ended up being a tight fit but I got everything in there (2 slot mode). I still had over heating problems and a 2tb 2.5 laying around so I went with that. When I have the time I might switch to three slot mode and try again. I want to say the issue had more to do with too much heat for the riser cable than anything else. My cpu and gpu thermals were fine but I kept getting an overheating audible warning.

This is the heatsink I used:

EKWB EK-M.2 NVMe Heatsink, Black https://www.amazon.com/dp/B073RHHYCM/ref=cm_sw_r_cp_apip_iYIjYdRpnufNa