EnzymeAD/Enzyme
Enzyme is a high-performance automatic differentiation compiler plugin for LLVM and MLIR that generates gradients for arbitrary functions.

Velocity · 7d
+0.6
★ / day
Trend
→steady
star history
Enzyme performs automatic differentiation on statically analyzable LLVM and MLIR code by replacing calls to __enzyme_autodiff with computed gradients. It integrates with popular ML frameworks including PyTorch and TensorFlow for gradient computation. The tool operates at the compiler level, allowing it to analyze and differentiate optimized code while maintaining performance that meets or exceeds state-of-the-art AD tools.