Improve code quality of `std.py` for Hessian and standard deviation computations.

[TianyuDu]

I found there is new documentation on Hessian computation here, we might want to improve the current code of Hessian computation with functorch.