I would like to use automatic differentiation to calculate gradients to function written in numpy.

I've come across a number of packages, including

But none of them seem to support things like numba and numexpr, which I'd normally use to accelerate my python code.

What packages do people use for this?

p.s. I know there's also stuff like tensorflow and pytorch, but I would like to keep my code in numpy

Your Answer


By clicking "Post Your Answer", you acknowledge that you have read our updated terms of service, privacy policy and cookie policy, and that your continued use of the website is subject to these policies.

Browse other questions tagged or ask your own question.