pytorch/test/distributed/optim
..
test_apply_optimizer_in_backward.py
test_named_optimizer.py
test_zero_redundancy_optimizer.py