Add GPU-accelerated ALS
Open
Nobody has claimed this yet.
components
performance
- Dominant language
- Python
- Stars
- 314
- Forks
- 77
- Avg merge
- 4d 6m
- Merged PRs (30d)
- 10
Description
Probably needs to be written in Numba, but we should be able to GPU-accelerate ALS. Possibly with conjugate gradient instead of Cholesky decomposition.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
The issue does not name any files or tests. Start by locating the existing ALS implementation, then investigate the proposed Numba GPU approach and the possible conjugate-gradient replacement for Cholesky decomposition; the work is done when ALS has GPU acceleration with its existing behavior preserved.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100