Add support for recalculate_users/items on the GPU

benfred / implicit

Fast Python Collaborative Filtering for Implicit Feedback Datasets

https://benfred.github.io/implicit/

MIT License

3.57k stars 612 forks source link

Add support for recalculate_users/items on the GPU #515

Closed benfred closed 2 years ago

benfred commented 2 years ago

Add GPU support for recalculating items and users factors at inference time. Also speed up the CPU versions by using the native solver rather than the python version.