[phdelightful]

If you can tolerate a .gov email address, there are high-performance computing, edge computing, and AI groups at DOE labs. My impression is that they're always starved for talent in these areas due to stiff competition from industry and lifestyle restrictions that come with a security clearance (being a US citizen, reduced international travel, no drugs other than alcohol and nicotine, your work may require on-site). Defense labs are probably similar.