Replace torch.nn.Module with metatensor.torch.learn.nn.Module
#938
Loading
torch.nn.Module with metatensor.torch.learn.nn.Module
#938