pytorch/torch/distributed/optim
Aaron Gokaslan 31715be72a [BE]: Update mypy to 1.11.2 (#133816)
Updates mypy to 1.11.1 to improve type inference

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133816
Approved by: https://github.com/ezyang
2024-09-16 19:44:11 +00:00
..
__init__.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
apply_optimizer_in_backward.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adadelta.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adagrad.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adam.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adamax.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adamw.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_rmsprop.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_rprop.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_sgd.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
named_optimizer.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
optimizer.py Revert "[BE] typing for decorators - _jit_internal (#131573)" 2024-07-28 03:29:32 +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][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
zero_redundancy_optimizer.py Pass torch.load(weights_only=) internally to avoid FutureWarning (#130663) 2024-07-16 01:24:38 +00:00
zero_redundancy_optimizer.pyi Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419) 2024-06-29 09:23:39 +00:00