I stumbled on this excellent article from 2015 that describes a really cool mechanism for tracing function execution in Python. This technique is used in the [autograd library](
https://github.com/HIPS/autograd), which is a precursor to [Jax](
https://github.com/google/jax), a competitor to Tensorflow and pytorch that's starting to gain traction.