trunk/d6f69da181966ba142cdc40c5d1a3adadf16f4d8: [dynamo][claude] Speedup tree_map_with_path (#174146)

A recent PyTorch commit optimized the `tree_map_with_path` function, reducing Dynamo compilation time from 12.2 seconds to 10.9 seconds for an internal Claude model—a 1.3 second speedup. The change follows the same specialization pattern as `tree.map` but maintains code readability. This performance improvement, approved by PyTorch maintainers, demonstrates how minor optimizations in core frameworks can yield measurable gains for large AI models.