pytorch

24294958 - [FSDP2][ez] Made typing more strict to avoid `cast` (#119985)

[FSDP2][ez] Made typing more strict to avoid `cast` (#119985) Pull Request resolved: https://github.com/pytorch/pytorch/pull/119985 Approved by: https://github.com/Skylion007, https://github.com/fegin ghstack dependencies: #118298