Apply PERF401 autofixes from ruff (#140980)
Summary:
* Automatically applies ruff rule 401. Turns loops into equivalent list comprehensions which are faster and do not leak the scope of the loop variables.
* list comprehensions not only often have better typing, but are 50+% faster than for loops on overhead. They also preserve length information etc and are better for the interpreter to optimize.
* Manually went back and made mypy happy after the change.
* Also fixed style lints in files covered by flake8 but not by pyfmt
X-link: https://github.com/pytorch/pytorch/pull/140980
Approved by: https://github.com/justinchuby, https://github.com/malfet
Reviewed By: izaitsevfb
Differential Revision: D66262948
fbshipit-source-id: 4d871761b25633da20bca8d1f37b9842144f2218
