Hi developers,
After I upgraded functorch from `v0.1.1` to `0.2.0`, I noticed a 25% performance regression when calculating hessian, please check the following benchmark result and the attached ben…
### System Info
```Shell
# Machine Learning and Deep Learning Libraries
torch==2.4.1 --index-url https://download.pytorch.org/whl/cu121 # Deep learning framework
torchvision==0.17.1 # Co…
## sway version
sway version v1.5
## log
[log](https://gist.github.com/plattfot/a8371fbe37501bc52a7bbdd823a0b265)
**Note** the log is quite big as sway was running for a few hours before it happ…