Forward AD formulas batch 2 (#57863)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57863
Test Plan: Imported from OSS
Reviewed By: zou3519
Differential Revision: D28387763
Pulled By: albanD
fbshipit-source-id: e1b60ab728bb05b9e3323ee0dc7e401aaf5b8817