Adding uses and definitions to Var

[erick-xanadu]

Hello,

I have read the autodidax's documents on JAX transforms and their connection to tagless-final style; however, I find writing transformations using def-use chains easier than implementing an interpreter with different semantics and re-evaluating the program. I was wondering if we could augment the class Var with uses and definition.

One would need to maintain the uses and definitions as the initial and final style transforms are executing, but it would also open the door for more traditional transforms. E.g., iterate over the JaxprEqns until a match is found and look at the definitions of the operands or the uses of the Var defined there and transform it.