Generalizing Neural Wave Functions