RecTools
RecTools copied to clipboard
gpu support for `ImplicitRanker`
Feature Description
use_gpu
argument to ImplicitRanker.rank
method that calls implicit.gpu.KnnQuery().topk
instead of implicit.cpu.topk.topk
when set to True.
Why this feature?
Allows for gpu inference for a range of models
Additional context
Described at https://github.com/MobileTeleSystems/RecTools/issues/199