This is a good idea. I had not considered hooks. Considering my next step, I would also like to modify the actual autograd backward() based on this recorded history. Ideally, it would just be more encapsulated to the particular Function instance.