Thanks a lot for the response.
The issue in removing the related (jit) errors is that every change I make is leading to it breaking somewhere else.
Anyways, one of my colleagues says that torchvision 0.4.0 should work fine. Trying that.