Just installed the Western Digital Black 1TB SN850 with this heak sink -
https://www.amazon.com/gp/product/B073RH58WN/ref=ppx_yo_dt_b_asin_title_o01_s01?ie=UTF8&psc=1
So far so good - read speed after format was 6547.75 MB/S