More precise resource usage tracking

[abidwael]

Using torch.profiler to track resource usage more precisely.

[github-actions[bot]]

Unit Test Results

         6 files  ±0         6 suites  ±0   2h 20m 21s :stopwatch: - 30m 21s   3 372 tests +1  3 294 :heavy_check_mark: +1    78 :zzz: ±0  0 :x: ±0  10 116 runs  +3  9 858 :heavy_check_mark: +3  258 :zzz: ±0  0 :x: ±0 

Results for commit b2a6a46c. ± Comparison against base commit bb5e2e88.

:recycle: This comment has been updated with latest results.

[abidwael]

Would it be hard to add a human-readable output? If not, adding the units for each of the reported metrics would help

@jppgks in PR #2372