If you see the Mac rebooting after waking up from sleep mode.
This problem will happen to all NVMe SSD drive including APPLE's original SSD.
We offer following solution to avoid this problem.
Please follow the steps below.
1. Download Transcend JetDrive ToolBox (Link:https://us.transcend-info.com/Support/Software-181/)
2. Open JetDrive ToolBox
3. Turn on System Optimize