Torch 2.2 and compile improvements

• Update torch version
• Add docs and test for checkpoint repair after compile
• Refactor mask merging logic for v2 encoder to reduce memory use
• Fix memory leak in v2 encoder during compile https://github.com/pytorch/pytorch/issues/107175