Claudi Lleyda Moltó

It is left as a draft PR, as I haven't had the chance to run the GPU code.

The performance results show that, the executions of both inference and training take less time. Most notably, from 90% of execution time is spent on the actual inference calculations of...

Sure, all of this is from metro16. For the profiles, the time measurements are off due to the profiling itself, what matters are the percentages. `aten::linear` and `aten::addmm` correspond to...

@raimis have you had the chance to look at this?