Hi, there is no way to convert a jit:script::Module to an nn::Module. However,  we should work on making it so that jit::script::Module can be used interchangeably. Can you comment on C++ API: Training and inference of torchscript modules on multiple GPU · Issue #46460 · pytorch/pytorch · GitHub 6 with your specific use case ? Thanks.