functorch not supported

[Pellarin]

functorch is deprecated and included in newest versions of torch as torch.func. A simple workaround is to change line 51 of align_scape to something like:

import torch.func as functorch