theseus
theseus copied to clipboard
Upgrade to PyTorch 1.11 to use vmap from functorch