我想使用自动微分来计算梯度到用 numpy 编写的函数。

我遇到了很多包，包括

• 自动毕业
• 切线
• 链轮

但是它们似乎都不支持像 numba 和 numexpr 这样的东西，我通常会用它们来加速我的 python 代码。

人们为此使用什么软件包？

ps 我知道还有 tensorflow 和 pytorch 之类的东西，但我想把我的代码保存在 numpy 中