Staff AI Engineer, SMAI | Micron Technology