Use `grad_transform` in `torch.autograd.Function` lookaside #4555

Job	Run time
check-package / init-store	6s
check-schema / schema	15s
precommit-run / pre-commit	35s
check-package / pkg-build (ubuntu-latest)	22s
check-package / pkg-check (ubuntu-latest, 3.10)	2m 43s
check-package / pkg-check (ubuntu-latest, 3.11)	2m 58s
check-package / pkg-check (macOS-latest, 3.10)	54s
check-package / pkg-check (macOS-latest, 3.11)	52s
check-package / pkg-check (windows-latest, 3.10)	3m 6s
check-package / pkg-check (windows-latest, 3.11)	3m 42s
check-package / pkg-guardian	1s
	15m 34s

Provide feedback