Thanks for the feedback. I am not familiar with the term PyTorch JIT’s fuser. Is this 3 where I should be reading from?
PS:
I just got these two useful links. I will go through it and try a second implementation.
Research to Production: PyTorch JIT/TorchScript Updates - Michael Suo
https://pytorch.org/docs/stable/jit.html 3