IntelLabs / IntelLabs/EquiTriton
Refactor and/or create new benchmark comparisons using the newer kernels up to l=10
Open
documentation
- Dominant language
- Python
- Stars
- 75
- Forks
- 7
- PR merge metrics
- No merged PRs in 30d
Description
Following #9, the first version of the kernels is not as dramatic in performance difference to `e3nn` + `torch.compile`. It would be good to see if the second versions, i.e. those that go up to `l=10` might see performance benefits.
Contributor guide
Research direction
Start by reading issue #9 and locating the existing kernel benchmark comparisons. Evaluate the newer kernel versions that support l=10 against e3nn with torch.compile; done means the benchmark comparisons are updated and clearly report whether the newer kernels improve performance.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning, performance
- Issue type
- Refactor
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 30/100