If you choose to grain fill then you should do so after staining. Be sure you use a grain filler that is compatible with your final finish. Should you use a colored grain filler then you also have the choice of either applying it directly over the stained wood (in which case it will alter the color of the stained wood) or seal the stained wood prior to applying filler.