@tom  If you could, I think it would be pretty awesome.
In my case, I would use it to resolve backward pass problems in erwins https://github.com/google-research/tiny-differentiable-simulator 1
I suspect some tooling would be helpful in this endeavor.  I’ll go at it the old fashioned way for a bit, and if it’s too hard, then perhaps I’ll see if I can hack my own checks.
Thanks for explaining the code, appreciated.