Can OpenMDAO co-operate with autograd or jax?

Viewed 108

Could the autograd or jax packages be used to generate the equivalent of analytic derivatives for OpenMDAO explicit components? i.e. something more accurate than finite differences (or perhaps more accurate or more general than the complex step method?) but without the work of manually computing and programming the analytic gradients?

I'm not an expert on either of these packages, but they seem to be designed for just this purpose.

0 Answers
Related