git clone https://github.com/dnbt777/tbug/
you must first log the tensors during a run. to do this, add the following to your code:
import tdiff
...
tdiff.capture(tensor, name="section_name.child_name", project="project_name_or_run_id_here")
you (probably) need to disable compilation for this to work (i.e. jit if in jax, or pytorch.compile if using pytorch)
