Kioxia has developed a new AI SSD that is designed to provide a secondary cache for Nvidia AI GPUs. The new drive comes with the manufacturer's XL-Flash that is SLC based and performs at over 10 million IOPS.
The Seagate FireCuda X1070 is a budget-minded SSD done mostly right. It performs well enough and has excellent support, but initial pricing is high. A Gen 4 SSD that you can just throw into any system and get the job done.