-
Notifications
You must be signed in to change notification settings - Fork 21.5k
vmap Basics
Richard Zou edited this page Dec 19, 2022
·
2 revisions
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 through the functorch whirlwind tour.
Read through https://github.com/pytorch/pytorch/blob/master/functorch/writing_batching_rules.md