RecTools
RecTools copied to clipboard
gpu support for `ImplicitRanker`
trafficstars
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