Of course, if you are using a contrasting colour grain filler for highlights, then you need to stain before.
Do you mean grain fill before?
Sanding back the grain filler is going to take the stain off along with it.