Replacing library imported in all sublibraries in Python

Question

I have a python virtual environment that uses some libraries. Many of those use numpy internally.

I now need to add gradient computations to the function