[Feature] An added function to change existing model definitions

[sarthakpati]

Hi,

Thank you for this work! I would like to propose a new function that can take existing models and then returns a new model type where the nn.Linear layer(s) have been changed to NdLinear layers.

Kind regards,
Sarthak

[alexreneau]

Great proposition! We're working on it :)