Annotate warmup for pytorch profiler

[michaelmckinsey1]

• This annotates the "warmup" region in the profile. See the below example

• This is a draft because I want to add this into the begin_code_region annotations that already exist for caliper, so there is not redundant code for annotations. If you look at the amount of annotations, it will make sense why this is a big issue. One thing that is annoying about the pytorch annotations is they are context-based, so any code regions need to be nested, which I think decreases readability.
• this PR also turns of the stack option, if we annotate all of the regions of interest it will be more succinct than showing the full stack