functorch icon indicating copy to clipboard operation
functorch copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

[POC] vmap for TensorDict

Open vmoens opened this issue 2 years ago • 0 comments

Makes the necessary changes to vmap.py to make Using TensorDict for functorch work

vmoens avatar Aug 13 '22 08:08 vmoens