Thanks for your answer, which saved me putting lots of sparse effort.
To be honest, I have not tried jit yet; but I need to know which path to take before going forward with it.
From your answer I can say, as I am using PyTorch, then, it’s better to use jit.