Enhance torch.vmap support from inside torch.compile (pytorch#116050)

This work rewrites vmap support in torch.compile by inlining most of
the frames into the existing FX graph. It also unlocks to PyTorch to
support features that were previously missing, such as keyword args.

Fixes: pytorch#114306

Pull Request resolved: pytorch#116050
Approved by: https://github.com/zou3519

.flake8

@@ -34,6 +34,8 @@ per-file-ignores =
     torch/utils/cpp_extension.py: B950
     torchgen/api/types/__init__.py: F401,F403
     torchgen/executorch/api/types/__init__.py: F401,F403
+    test/dynamo/test_higher_order_ops.py: B950
+    torch/testing/_internal/dynamo_test_failures.py: B950
 optional-ascii-coding = True
 exclude =
     ./.git,