iree-org/wave

View on GitHub

Wave: Python Domain-Specific Language for High Performance Machine Learning

Stars: 59
Forks: 32
Open beginner issues: 2
Indexed issues: 93

Dominant language: Python
License: Apache-2.0
Last GitHub push: Apr 27, 2026
Latest indexed: Sep 17, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: good first issue help wanted
PR merge metrics: No merged PRs in 30d

Beginner friendly Everything

Open All

Introduce GatherToLDS MLIR operation Open

Difficulty 4/5 3-5 days Newbie friendliness 48/100

iree-org/wave#1290 · Opened Apr 10, 2026
[water] check dynamic dimension support for mxfp4 Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

iree-org/wave#1268 · Opened Apr 7, 2026
[water] scheduling pass Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#1267 · 1 comment · Opened Apr 7, 2026
[wave] NSA: SGLang / inference backend integration Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 32/100

iree-org/wave#1263 · Opened Apr 4, 2026
[wave] NSA: water dialect lowering for NSA composite ops Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#1262 · Opened Apr 4, 2026
[wave] NSA: performance benchmarks on MI350 Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 30/100

iree-org/wave#1261 · Opened Apr 4, 2026
[wave] NSA: gradient correctness tests (backward pass) Open

enhancement nsa

Difficulty 4/5 3-5 days Newbie friendliness 45/100

iree-org/wave#1260 · Opened Apr 4, 2026
[wave] NSA: numerical correctness tests vs dense attention reference Open

enhancement nsa

Difficulty 4/5 3-5 days Newbie friendliness 45/100

iree-org/wave#1259 · Opened Apr 4, 2026
[wave] NSA: GQA-aware memory layout & tiling for MI350 cache hierarchy Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 30/100

iree-org/wave#1258 · Opened Apr 4, 2026
[wave] NSA: LDS optimization for block-gather in selection attention Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1257 · Opened Apr 4, 2026
[wave] NSA: CDNA4 wavefront scheduling for sparse gather patterns Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#1256 · Opened Apr 4, 2026
[wave] NSA: end-to-end training integration with autograd Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1255 · Opened Apr 4, 2026
[wave] NSA: gating parameter gradients & top-k STE backward Open

enhancement nsa

Difficulty 4/5 3-5 days Newbie friendliness 45/100

iree-org/wave#1254 · Opened Apr 4, 2026
[wave] NSA: compressed attention backward kernel Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 38/100

iree-org/wave#1253 · Opened Apr 4, 2026
[wave] NSA: selection attention backward kernel (dQ, dK, dV) Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1252 · Opened Apr 4, 2026
[wave] NSA: end-to-end inference pipeline Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1251 · Opened Apr 4, 2026
[wave] NSA: gated output combination kernel Open

enhancement nsa

Difficulty 4/5 3-5 days Newbie friendliness 48/100

iree-org/wave#1250 · Opened Apr 4, 2026
[wave] NSA: sliding window attention forward kernel Open

enhancement nsa

Difficulty 4/5 3-5 days Newbie friendliness 48/100

iree-org/wave#1249 · Opened Apr 4, 2026
[wave] NSA: selection attention forward kernel Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1248 · Opened Apr 4, 2026
[wave] NSA: top-k block selection kernel Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1247 · Opened Apr 4, 2026
[wave] NSA: compressed attention forward kernel Open

enhancement nsa

Difficulty 4/5 3-5 days Newbie friendliness 35/100

iree-org/wave#1246 · Opened Apr 4, 2026
[wave] NSA: token compression kernel (mean pooling KV blocks) Open

enhancement nsa

Difficulty 4/5 3-5 days Newbie friendliness 48/100

iree-org/wave#1245 · Opened Apr 4, 2026
[wave] NSA: architecture mapping & design doc for MI350 Open

enhancement nsa

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#1244 · Opened Apr 4, 2026
[wave] DeepSeek Native Sparse Attention (NSA) kernels for MI350 Open

nsa tracker

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#1243 · Opened Apr 4, 2026
[water] Use InferIndexExprsOpInterface descriptions in ScaledMmaOp EPT propagation Open

Difficulty 3/5 1-2 days Newbie friendliness 48/100

iree-org/wave#1189 · Opened Mar 25, 2026
[water] harden verification of hyperparams and vector shapes to only admit i64 entries Open

iree-org/wave#1179 · 1 reaction · 2 assignees · Opened Mar 24, 2026
link water/waveasm with lld in setup.py Open

Difficulty 2/5 1-3 hours Newbie friendliness 64/100

iree-org/wave#1178 · Opened Mar 24, 2026
[water] first-class dict-like attribute with WaveSymbolAttr as key Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#1142 · 1 comment · Opened Mar 18, 2026
Scalar global memref.load/store hardcodes SRD s[8:11], so multi-buffer kernels alias arguments Open

Difficulty 3/5 1-2 days Newbie friendliness 45/100

iree-org/wave#1117 · Opened Mar 12, 2026
[water] investigate water dialect connection to assembly dialects Open

iree-org/wave#1083 · 1 assignee · Opened Mar 9, 2026
[wave<->water] create a fuzzer for matmuls and attentions Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1082 · Opened Mar 9, 2026
output memref sizes are bogus and excessively large Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

iree-org/wave#1064 · Opened Mar 6, 2026
[water] consider adding a top wave.kernel operation Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#1049 · Opened Mar 5, 2026
Outstanding Issues with 4-wave WaveASM Kernel Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

iree-org/wave#1005 · Opened Feb 27, 2026
[water] consider using water bindings to run passes Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#984 · Opened Feb 26, 2026
[water] even more persistent water emitters Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

iree-org/wave#983 · Opened Feb 26, 2026
mark_hardware_transpose_candidates breaks elements-per-thread invariant Open

iree-org/wave#982 · 2 comments · 1 assignee · Opened Feb 26, 2026
The semantics of strides in IndexSequence is inconsistent Open

iree-org/wave#956 · 1 assignee · Opened Feb 23, 2026
[water] Use WaveSymbolMapping for index mappings Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

iree-org/wave#943 · Opened Feb 20, 2026
Reshape target_vector_shape semantics are inconsistent Open

iree-org/wave#942 · 2 comments · 2 assignees · Opened Feb 20, 2026
tag attribute Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

iree-org/wave#923 · Opened Feb 19, 2026
[wave2water] block sum reduction fails to convert Open

iree-org/wave#904 · 1 assignee · Opened Feb 18, 2026
sum (and potentially other) block reduction throws random type errors Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

iree-org/wave#903 · 1 comment · Opened Feb 18, 2026
`apply_expr` typing is broken for comparisons Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

iree-org/wave#901 · Opened Feb 18, 2026
consider having a WaveExprAttr that is not a list Open

Difficulty 5/5 Over a week Newbie friendliness 30/100

iree-org/wave#898 · Opened Feb 18, 2026
attempt to solve the Type/TypeConstraint conundrum locally Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#889 · Opened Feb 16, 2026
[water] make index attribute verifier check that it provides index mappings for all symbols used in the operation Open

iree-org/wave#878 · 1 reaction · 2 assignees · Opened Feb 12, 2026
in-thread transpose creates incorrect reshape operations Open

iree-org/wave#873 · 1 assignee · Opened Feb 11, 2026
Unify attribute conversion registry between FX → MLIR and MLIR → FX directions Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

iree-org/wave#872 · Opened Feb 11, 2026
Model `start` and `condition` in MLIR `IterateOp` Open

Difficulty 3/5 1-2 days Newbie friendliness 48/100

iree-org/wave#870 · Opened Feb 11, 2026

iree-org/wave

Get new issues in your inbox