SN850 1TB disconnects after idle

Look here, the problem seems to be very similar:

WD_Black SN850 not detected by BIOS on bootup from power-off - WD SSD Products / WD SSD Drives & Software - WD Community