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

Slow compile time for baroclinic model Open

EnzymeAD/Enzyme-JAX#677 · 7 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
[All2All] Generalize triple concat Open

EnzymeAD/Enzyme-JAX#691 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Communication standing issue Open

EnzymeAD/Enzyme-JAX#695 · 21 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Reshape of DUS Open

EnzymeAD/Enzyme-JAX#696 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Rotate of Slice / Wrap of Slice Open

EnzymeAD/Enzyme-JAX#700 · 0 comments · 0 reactions · 1 assignee · Indexed Sep 15, 2026
Select to concat or dynamic_update_slice Open

EnzymeAD/Enzyme-JAX#703 · 3 comments · 0 reactions · 1 assignee · Indexed Sep 15, 2026
concate of rotate -> rotate of concats Open

EnzymeAD/Enzyme-JAX#715 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Weird dus update Open

EnzymeAD/Enzyme-JAX#717 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
While of wrap or extend as returned value -> wrap of while Open

EnzymeAD/Enzyme-JAX#745 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`LLVM ERROR: Failed to infer result type(s)` with sharding Open

EnzymeAD/Enzyme-JAX#755 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Not recognizing extend: Open

EnzymeAD/Enzyme-JAX#758 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Crash during jit compile conversion from stablehlo -> hlo proto -> hlo Open

EnzymeAD/Enzyme-JAX#763 · 3 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Last remaining all-gathers Open

EnzymeAD/Enzyme-JAX#770 · 6 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`all-reduce`s in sharded baroclinic model Open

EnzymeAD/Enzyme-JAX#778 · 4 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`UNIMPLEMENTED: PartitionId instruction is not supported for SPMD partitioning since the meaning is ambiguous` Open

EnzymeAD/Enzyme-JAX#789 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Simplify comparison Open

EnzymeAD/Enzyme-JAX#811 · 4 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
OOM in XLA when compiling functions Open

EnzymeAD/Enzyme-JAX#823 · 8 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`Check failed: Product(new_dimensions) == iota_->num_elements() (2 vs. 4)` Open

EnzymeAD/Enzyme-JAX#828 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`error: 'stablehlo.multiply' op requires compatible types for all operands and results` Open

EnzymeAD/Enzyme-JAX#839 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`error: 'stablehlo.concatenate' op failed to infer returned types` Open

EnzymeAD/Enzyme-JAX#840 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Generalize Batch-Op Open

EnzymeAD/Enzyme-JAX#860 · 3 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Simplify Reduce(Reduce(...)) Open

EnzymeAD/Enzyme-JAX#867 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Common Linear Algebra Routines Open

EnzymeAD/Enzyme-JAX#886 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`Concat(Reshape(Reduce(...)))` Open

EnzymeAD/Enzyme-JAX#892 · 9 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
concat of reshapes when applied along 0-dim Open

EnzymeAD/Enzyme-JAX#905 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
[arith] Missing canonicalize Open

EnzymeAD/Enzyme-JAX#913 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
[arith] more simplification Open

EnzymeAD/Enzyme-JAX#920 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Transpose of gather Open

EnzymeAD/Enzyme-JAX#937 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`triangular_solve` optimizations Open

EnzymeAD/Enzyme-JAX#940 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
raise `lu` + `lu_solve` into `linalg.linsolve` Open

EnzymeAD/Enzyme-JAX#941 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
stablehlo.if constprop Open

EnzymeAD/Enzyme-JAX#955 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Check which patterns support dynamic shapes Open

EnzymeAD/Enzyme-JAX#962 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Default example in README does not work Open

EnzymeAD/Enzyme-JAX#964 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
`lu` optimizations Open

EnzymeAD/Enzyme-JAX#982 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
ONNX fusion passes Open

EnzymeAD/Enzyme-JAX#986 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
One Hot Array Optimization Patterns Open

EnzymeAD/Enzyme-JAX#991 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Attention Ops Open

EnzymeAD/Enzyme-JAX#994 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
fft optimization patterns Open

EnzymeAD/Enzyme-JAX#1003 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
conj optimization patterns Open

complex

EnzymeAD/Enzyme-JAX#1005 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
ConcatConcatToDus crashes Open

EnzymeAD/Enzyme-JAX#1023 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Getting started Open

EnzymeAD/Enzyme-JAX#1024 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
GPU Codegen Open

EnzymeAD/Enzyme-JAX#1027 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
concatenate of same value is reshape -> bcast -> reshape Open

EnzymeAD/Enzyme-JAX#1032 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 15, 2026
macos tag failure Open

EnzymeAD/Enzyme-JAX#1037 · 2 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Host/device signature mismatch Open

EnzymeAD/Enzyme-JAX#1066 · 5 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Triangular Matrix Multiply Open

EnzymeAD/Enzyme-JAX#1067 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
FFT of Real Tensor to RFFT with Concat Open

EnzymeAD/Enzyme-JAX#1068 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Structured Matrix / Tensors Detection Open

EnzymeAD/Enzyme-JAX#1072 · 1 comment · 1 reaction · 0 assignees · Indexed Sep 15, 2026
Reuse common power op Open

EnzymeAD/Enzyme-JAX#1080 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 15, 2026
Matrix Multiply Chain CSE Open

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

EnzymeAD/Enzyme-JAX

Get new issues in your inbox