Need help with Umlaut or tracing in general?

[dfdx]

Coming from this thread, I noticed you had experimented with Umlaut previously and presumably moved to your own tracer. If you still work on this project and need help with either of these, I'd be happy to offer it.

For example, in the Umlaut test you might have been looking for trace!(::Tracer, v_fargs) to recursively trace JVPs like I did it for VJPs here. In the own tracer, I wonder why you decided to go with the plain CodeInfo instead of IRCode since I found the later to be much more pleasant to work with.

In any case, I'm looking forward to see what your experiments end up with!