Page Maintainers: @zou3519

• Understand what vmap is
• Understand how vmap is implemented with Tensor Subclassing
• Understand (as a PyTorch developer) how to add missing support for an operation with vmap

Read throughthe functorch whirlwind tour.

Read throughhttps://github.com/pytorch/pytorch/blob/master/functorch/writing_batching_rules.md

Unit 8: function transforms/Training Loops (Optional) -vmap-onboarding-lab