gistfile1.py

import torch

class SomeMod(torch.jit.ScriptModule):
    @torch.jit.script_method
    def _unpack(self):
        return torch.zeros(3, 4)

    @torch.jit.script_method
    def forward(self):
        return torch.zeros(3, 4)

class Mod(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.sm = SomeMod()

    def forward(self, x):
        return x + self.sm()

traced = torch.jit.trace(Mod(), (torch.rand(3, 4),))

print(traced)
print(traced.sm)
print(traced.sm._has_method('_unpack'))

===

TracedModule[Mod](
  (sm): TracedModule[SomeMod]()
)
TracedModule[SomeMod]()
False