My understanding is you still need to use EAD

Some officers may process with "AOS" on I-94, some may not. Unless the regulation clearly allows in this situation.
EAD is work document and surely is well accepted for process of SSN application.