I think it also helps to think of it in a community or society sense too. If a community has a small number of neurodiversity they are better for it, but if everyone had it then there would be problems. For example depression could be thought of as a kind of introspection and are good at noticing negative things, having that kind of person in a group noticing problems and prompting change would be useful. Those with ADHD would be better noticing/finding things, again not great at things like sitting through boring meetings but useful to a community. Even most pre modern societies had a role like that of a shaman or a spiritual leader who was often schizophrenic, and this again would be useful to the community on the whole.
So my theory is that some of our evolution is not at the individual level but the societal.
So my theory is that some of our evolution is not at the individual level but the societal.