I got around that NVMe bug by installing OEM drivers prior to upgrading.
The issue is stornvme.sys has a bug that causes the drive to freeze. The WHEA_UNCORRECTABLE_ERROR code is the result of it triggering detection of a hardware fault within the PCI interface, causing the system to bugcheck. The driver I use is a Phision branded driver package, based on the Open Fabrics Alliance NVM Express driver version 1.5, which they simply got a WHQL signature on the reference binary and packaged in their own .inf file.