Help integrating Enzyme for StableHLO

[Joel Berkeley]

Hi all,

I'm writing a research proposal for a postdoc involving the distributed computing API of StableHLO, and it would be very useful to be able to include autodiff in the work. I anticipate I would need help to integrate enzyme, and wonder if anyone here would have the time to discuss this?

Regards,

Joel Berkeley

[William Moses]

Hey Joel,

We’ve actually been working on adding Enzyme support to StableHLO quite a bit the past few months.

The code for this is in the Enzyme-JaX repo and tests are here: 

https://github.com/EnzymeAD/Enzyme-JAX/tree/main/test/lit_tests/diffrules/stablehlo

I’d love to chat more about this, want to find a time next week?

Cheers,

Billy

--
You received this message because you are subscribed to the Google Groups "Enzyme AD" group.
To unsubscribe from this group and stop receiving emails from it, send an email to enzyme-dev+...@googlegroups.com.
To view this discussion on the web visit https://groups.google.com/d/msgid/enzyme-dev/58d98d90-f6fb-4c6e-b679-efb222e73eddn%40googlegroups.com.

[Joel]

Thanks. I'm around all next week except weds eve.

For now, would you be able to say whether you've finished implementing AD for all StableHLO ops, or if that's work in progress? No need to go into depth.