How does backward work with JIT module?

jit
1

I’ve read README.md about JIT at (https://github.com/WyldeCat/pytorch/blob/master/torch/csrc/jit/README.md).

There is a great explanation about how JIT handles a module’s forward method.

But I can’t find any explanation about how backward pass works.

So, I want to ask whether JIT involves in backward pass or not and is there any way to print backward pass’s graph like JIT module’s forward graph.

Thanks in advanced.

2 Likes