That could be addressed with a small NVMe heatsink. They're available and their use is advised already for NAND PCIe 4.0 and 5.0 hardware, but they would fit the Optane use just as well.