This is an implementation of BM25 that I have been building for the last few months. It builds on top of the optimizations implemented in BM25S, via category theory based abstractions and scaffolding.
https://www.rajeshrs.in/blog/ai-explorations/2025-12-24-vajr...