I don't know much about the reality of 'baby brain', but in the study under discussion the researchers tested cognitive skills like memory, verbal skills, and working memory, and there was no significant change over time. It looks like the after-pregnancy tests were given late enough that kids were probably sleeping through the night (or at least in 5-6 hr chunks).