EnzymeAD

EnzymeAD/Enzyme-JAX

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

10 good first / help-wanted issues · MLIR · last activity Aug 25, 2026

130 stars 48 forks 130 watchers MLIR Other
View on GitHub
10 Open Issues Need Help Last updated: Aug 25, 2026

Open Issues Need Help

View All on GitHub
Tracking issue for missing HLO derivatives 1 day ago
good first issue help wanted
EnzymeAD/Enzyme-JAX
130

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

MLIR
Tracking issue for missing Batch Op Interface about 2 months ago
good first issue help wanted
EnzymeAD/Enzyme-JAX
130

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

MLIR
Missing autodiff rules for stablehlo.case: reverse fails with “could not compute the adjoint”, forward with “RegionBranchOpInterface not implemented” 2 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR
`x != false` and others 5 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR
`concat(a[n-1:n], a[n-2:n-1], ...., a[0:1])` to `reverse` 5 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR
AffineCFG: MoveSIToFPToAffine causes infinite loop when sitofp operand is a valid symbol in scope 5 months ago
help wanted
EnzymeAD/Enzyme-JAX
130

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

MLIR
Missing reduce constant prop 5 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR
ConvertParallelToGPU1 pass crashes 5 months ago
help wanted
EnzymeAD/Enzyme-JAX
130

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

MLIR
Extend DotGeneralSimplify to handle Broadcasted Scalars 5 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR
unroll while loop with known iteration count of 1 12 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR