Now that Enzyme is supported, it looks like the next big step in AD support is to get a Zygote-based extension and `DifferentiationMethod` implemented.