Keyword search (or something similar in concept, like bm25) would typically be first stage, rather than second, since it can be done with an inverted index.

I'm aware of that. My question was around using it in the ranking algorithm.