Any IO grad programs with a cognitive focus?

I am graduating with a BA in psych this fall. Until this summer I was determined to get a PhD in cognitive psychology with a focus on memory. After months of reflecting (I hate academia + no jobs), I decided that with my research skills and love for psychology, to get a degree in IO psychology. Either masters or doctorate, any programs out there that are cognitive focused?