vitamin B12 and magnesium are important for cognitive function.
The other B vitamins and vitamin c are also important. Imo it isn't a good idea to take plenty of B6 without the other B vitamins. Choline and carnitine may also be helpful, as well as fish oil.