Hi,
I think the inputs is expected to be a tuple, so you should pass (y,) no? Not sure if that changes anything though.
Have you checked that in the code sample you gave above, if you change torch.jit.trace(torch_model, y) with torch_model(y) it works fine?