EnzymeAD

EnzymeAD/Enzyme-JAX

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

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 Batch Op Interface 24 days 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” 26 days ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR
`x != false` and others 3 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` 4 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 4 months ago
help wanted
EnzymeAD/Enzyme-JAX
130

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

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

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

MLIR
ConvertParallelToGPU1 pass crashes 4 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 4 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR
Tracking issue for missing HLO derivatives 5 months ago
good first issue help wanted
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 11 months ago
good first issue
EnzymeAD/Enzyme-JAX
130

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

MLIR