openmdao - OpenMDAO 可以与 autograd 或 jax 合作吗？

Question

是否可以使用 autograd 或 jax 包为 OpenMDAO 显式组件生成等效的解析导数？即比有限差分更准确（或者可能比复杂步骤方法更准确或更通用？）但没有手动计算和编程分析梯度的工作？

我不是这些软件包的专家，但它们似乎就是为此目的而设计的。