EnzymeAD/Enzyme-JAX

View on GitHub

Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.

Stars: 131
Forks: 53
Open beginner issues: 0
Indexed issues: 291
Avg merge: 1d 10h
Merged PRs (30d): 193

Dominant language: MLIR
License: No license data
Last GitHub push: Sep 14, 2026
Latest indexed: Sep 15, 2026
Contributing guide: No contributing guide
Code of conduct: No code of conduct
Beginner labels: good first issue help wanted

Open All

291 open issues indexed

Lower Linear Algebra factorizations to StableHLO Open

EnzymeAD/Enzyme-JAX#1083 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Missing reduce constant prop Open

good first issue

EnzymeAD/Enzyme-JAX#1084 · 3 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Rework ReshapeBroadcast passes to reuse Helper functions Open

EnzymeAD/Enzyme-JAX#1090 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Transpose Elementwise always if all branches can remove transpose Open

EnzymeAD/Enzyme-JAX#1094 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
[DoWhile] Add in then region blocks while2for Open

EnzymeAD/Enzyme-JAX#1135 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
slice of DUS with subset indices can be replaced with a slice of update Open

EnzymeAD/Enzyme-JAX#1143 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Pass to convert top_k with k = 1 to a reduction Open

EnzymeAD/Enzyme-JAX#1144 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
How to handle -Werror errors? Open

EnzymeAD/Enzyme-JAX#1147 · 12 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Failed while2for Open

EnzymeAD/Enzyme-JAX#1158 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 15, 2026
maxi related simplifications Open

EnzymeAD/Enzyme-JAX#1164 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Missing canonicalizations (during canonicalize-scf-for) Open

EnzymeAD/Enzyme-JAX#1170 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
EnzymeJaX python error Open

EnzymeAD/Enzyme-JAX#1191 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
EnzymeMLIR: Generalize scatter reverse mode (to handle scatter_add) Open

EnzymeAD/Enzyme-JAX#1193 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Add of selects with same conditions -> select of adds Open

EnzymeAD/Enzyme-JAX#1202 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
slice of single use rotate -> rotate slice Open

EnzymeAD/Enzyme-JAX#1203 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
subtract wrap -> reduce window Open

EnzymeAD/Enzyme-JAX#1206 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
concat reshape slice -> dus reshape Open

EnzymeAD/Enzyme-JAX#1207 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`unexpected unnamed parameters` warning in GB-25 code Open

EnzymeAD/Enzyme-JAX#1213 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
rev diff of `stablehlo.dynamic_update_slice` is broken for complex tensors Open

EnzymeAD/Enzyme-JAX#1211 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Multi dus simplify Open

EnzymeAD/Enzyme-JAX#1216 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
reassociate fmul with constants Open

EnzymeAD/Enzyme-JAX#1220 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Concat of many same argument -> broadcast Open

EnzymeAD/Enzyme-JAX#1227 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Partial concat vectorization Open

EnzymeAD/Enzyme-JAX#1236 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Reverse removal Open

EnzymeAD/Enzyme-JAX#1237 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
failure to raise - terminator with potential side effects Open

EnzymeAD/Enzyme-JAX#1254 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
lower `enzymexla.linalg.qr` to `enzymexla.lapack.geqrf/geqrt` Open

lapack

EnzymeAD/Enzyme-JAX#1261 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
fuse `lapack.orgqr` + `stablehlo.dot_general` into `lapack.ormqr` Open

lapack

EnzymeAD/Enzyme-JAX#1262 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
lower `lapack` ops to `stablehlo` Open

lapack

EnzymeAD/Enzyme-JAX#1264 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
implement sharding rules for `lapack` ops Open

lapack sharding

EnzymeAD/Enzyme-JAX#1265 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
batch `lapack` routines Open

EnzymeAD/Enzyme-JAX#1267 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
diff rule for `lapack.geqrf`, `lapack.geqrt` Open

lapack

EnzymeAD/Enzyme-JAX#1268 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
support dynamic-size matrices on lowering of `lapack` routines Open

dynamism lapack

EnzymeAD/Enzyme-JAX#1269 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
implement `(he|sy)ev`, `(he|sy)evd` lapack routines for hermitian/symmetric eigendecomposition Open

EnzymeAD/Enzyme-JAX#1270 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
[CI] Failed jaxmd test Open

EnzymeAD/Enzyme-JAX#1289 · 3 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
MLIR error with while2for / mld Open

EnzymeAD/Enzyme-JAX#1301 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Cast `i64` indexing to lower precision automatically if possible Open

EnzymeAD/Enzyme-JAX#1303 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
conditionally perform convert(concat) -> concat(convert) Open

EnzymeAD/Enzyme-JAX#1340 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
broadcast(convert(...)) to convert(broadcast()) based on types Open

EnzymeAD/Enzyme-JAX#1350 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
remove unnecessary intermediate insertdims Open

EnzymeAD/Enzyme-JAX#1355 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
DUS(splat or broadcast, update) -> pad(update, 0-dim splat) Open

EnzymeAD/Enzyme-JAX#1362 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Auto-Batching MLIR Open

EnzymeAD/Enzyme-JAX#1364 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
LowerExtend with `lhs > size(x, dim)` Open

bug

EnzymeAD/Enzyme-JAX#1370 · 4 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
stablehlo while with non-affine bound Open

EnzymeAD/Enzyme-JAX#1382 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
missing all_finite simplifications Open

EnzymeAD/Enzyme-JAX#1387 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
continuous benchmarking Open

EnzymeAD/Enzyme-JAX#1390 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 15, 2026
mincut takes forever then crashses Open

EnzymeAD/Enzyme-JAX#1426 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
gather of setindex scatter Open

EnzymeAD/Enzyme-JAX#1428 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Mark ptrs as `llvm.nocapture` Open

EnzymeAD/Enzyme-JAX#1442 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`PadDotGeneral` is busted Open

EnzymeAD/Enzyme-JAX#1445 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Preserve arg attributes when lowering kernel (gpu) Open

EnzymeAD/Enzyme-JAX#1451 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026

EnzymeAD/Enzyme-JAX

Get new issues in your inbox