I keep on having issues with my WD SN850 dropping offline and coming back R/O but it still fails to produce any reads and my system still crashes. I've applied all the pcie power management fixes on the kernel command line and still having problems. That sounds similar to what you describe. It starts off with vscode disappearing, then UI locks, and then it dumps me to a virtual console spitting kernel errors for a bit.

The nvme drive itself reports no media errors so I really do think it's a system side problem with the pcie controllers...