https://fairytalefashionheadband.blogspot.com/2023/08/eager-execution-gradient-computation.html