Implement torch.profiler

[javak87]

To profile the code in terms of memory, torch.profiler needs to be implemented.

[javak87]

assign me

[github-actions]

Issue assigned to @javak87.