Hi, does Flyte support serializing

torch.ScriptModule

? (as in/output of Task) Getting:

Failed to convert outputs of task '{snip}.train_model' at position 0:
  Tried to serialize object __torch__.{snip} which does not have a __getstate__ method defined!

You will need to add a transformer For it