>My understanding is that AWS KMS uses AWS designed HSMs

That's my take as well reading about how they handle firmware (sounds like they're using their own chips, presumably similar to how they use other hardware acceleration and offload)