Rodrigo,
Thanks, yes haha i’ll add a license right now.
For features:

GPU support is easy I’ll implement it now, but i’ll look at yours if you already did it.
pretrained is not my area, but happy to accept your code
yes, accuracy logging and other metrics are definitely on to-do… will prob implement accuracy in the next few days

Note that a lot of the code is changing rapidly (for instance, just removed - th_gather2d/th_gather3d

in favor of -  th_gather_nd - … and made -  th_meshgrid - work for any number of dimensions (e.g. th_meshgrid(2,3,4)) , and have been adding a lot to the transforms. Hopefully the SuperModule will remain quite stable though so definitely happy to have contributions there.