UserWarning: We've integrated functorch into PyTorch. As the final step of the integration, functorch.vmap is deprecated as of PyTorch 2.0 and will be deleted in a future version of PyTorch >= 2.3. Please use torch.vmap instead; see the PyTorch 2.0 release notes and/or the torch.func migration guide for more details https://pytorch.org/docs/master/func.migrating.html
functorch is used:
[ ] To compute _UU in observable/UU.py functorch.vmap(ensemble.Action.FermionMatrix.UU)(ensemble.configurations)
[ ] To compute S in observable/action.y functorch.vmap(ensemble.Action)(ensemble.configurations)
[ ] To compute Contact_bosonic in observable/contact.py, functorch.vmap(S_dual)(ensemble.configurations)
Running with pytorch 2.0.1 I get
functorch is used:
_UUin observable/UU.pyfunctorch.vmap(ensemble.Action.FermionMatrix.UU)(ensemble.configurations)Sin observable/action.yfunctorch.vmap(ensemble.Action)(ensemble.configurations)Contact_bosonicin observable/contact.py,functorch.vmap(S_dual)(ensemble.configurations)