RFC: add support for batched matmul / einsum / batched tensordot
Nobody has claimed this yet.
Assessment
- Difficulty
- 5/5
- Estimated time
- Over a week
- Newbie friendliness
- 25/100
- Issue type
- Feature
- Clarity
- Needs clarification
- Activity status
- Stale
- Domain
- data
Research direction
The issue names no files, tests, or entry points. Start by reviewing the array API standard and comparing the proposed batched_matmul, batched tensordot, and einsum approaches; done means selecting and specifying a supported operation rather than leaving multiple alternatives open.
Written by the indexing model from the issue text.
Description
Currently I don't see a way to implement einsum or batched matmul using the standard. There is a way with explicit loop, but that does not count.
Some ways this can be addressed:
- minimal implementation would include batched_matmul (signature
bij, bjk -> bik). Other operations can be reduced to it with reshapes. This may include additional copies - modification of tensordot to include batched dimensions thus should be more efficient, but AFAIK this method is not surfaced in any major framework
- include einsum in the standard
- Dominant language
- Python
- Stars
- 281
- Forks
- 52
- PR merge metrics
- No merged PRs in 30d
Contributor guide
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.
More from data-apis/array-api
-
bug Maintenance Narrative Content
Difficulty 1/5 1-3 hours Newbie friendliness 88/100
-
Difficulty 2/5 1-3 hours Newbie friendliness 70/100
-
Difficulty 5/5 Over a week Newbie friendliness 35/100
-
Maintenance
-
Difficulty 5/5 Over a week Newbie friendliness 25/100
All issues in data-apis/array-api
Similar issues
-
bug
Difficulty 2/5 1-3 hours Newbie friendliness 90/100
learningequality/ricecooker#747 ·
-
Difficulty 2/5 1-3 hours Newbie friendliness 68/100
BSData/horus-heresy-3rd-edition#3171 ·
-
enhancement
Difficulty 2/5 1-3 hours Newbie friendliness 72/100
-
Difficulty 2/5 1-3 hours Newbie friendliness 76/100
run-llama/llama_index#23199 ·
-
Difficulty 2/5 1-3 hours Newbie friendliness 84/100
KhronosGroup/glTF-Blender-IO#2769 ·