pytorch/torch/distributed/optim
Aaron Gokaslan edd640a95a [BE][Ez]: Use itertools.chain.from_iterable when possible (#148190)
Often makes the code more readable, more efficient, and adds support for infinite iterables.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148190
Approved by: https://github.com/jansel, https://github.com/malfet
2025-03-06 20:37:06 +00:00
..
__init__.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_deprecation_warning.py Fix unintended deprecation warning in torch.distributed.optim (#140889) 2024-11-18 02:34:51 +00:00
apply_optimizer_in_backward.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
functional_adadelta.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adagrad.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adam.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adamax.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adamw.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_rmsprop.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_rprop.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_sgd.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
named_optimizer.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
optimizer.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
post_localSGD_optimizer.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
utils.py [BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408) 2025-02-04 19:07:04 +00:00
zero_redundancy_optimizer.py [BE][Ez]: Use itertools.chain.from_iterable when possible (#148190) 2025-03-06 20:37:06 +00:00
zero_redundancy_optimizer.pyi [BE] Upgrade to mypy 1.14 (#145966) 2025-03-04 20:58:26 +00:00