This package boosts a sparse matrix multiplication followed by selecting the top-n multiplication
pip install sparse-dot-topn