How to differentiate StableHLO with Enzyme-JAX from C++? #195
Open
Description
I'm trying to differentiate a StableHLO mlir::ModuleOp
, and I'm lost. I'm starting by trying to interface with Enzyme-JAX, but I've noticed that the only source file exported by bazel is enzymexlamlir-opt.cpp
. I did try to use registerStableHLODialectAutoDiffInterface
but that's not exported. What's the recommended usage of Enzyme-JAX from C++?
Metadata
Assignees
Labels
No labels
Activity