Solved by tom in post #2 
              
              
                What happens is that in the second run you get the JIT autodiffed abs and that has been broken when complex support was added to the derivative of backward. Sadly, it seems that while the autograd that you get on the first run had been updated 10 months ago, the JIT has not been getting as much love…