I’ve been looking at the eeprom versions of the Raspberry 5, and I found this
2024-01-24: NVMe boot fix for WD NVMe (latest)
Add a workaround for an issue seen when booting with WD Blue SN550 NVMe SSD
This seems to solve the problems with the WD drives, so an eeprom update to version 1/24 or higher should be enough to be able to use them