Hi,
I have been following the development of the JIT mode in PyTorch and using it for some of my own research. What I want it to be able to extract a graph (forward and backward) and run some custom passes on it. I have been using the jit.trace till now, but it turns out that gradients for things like sigmoid etc are not traced properly because of ( https://github.com/pytorch/pytorch/blob/master/torch/csrc/jit/tracer.h#L155 58 ). I was also looking at the more recent JIT script, but it turns out it only supports a small set of operations ATM. I just wanted to understand what is vision for JIT in PyTorch. Will the JIT script be expanded to cover all operations ? Will the tracing mechanism still be supported and developed further ?
Thanks and Regards
Tapan Chugh