116     def test_optim_infos_do_not_specify_global_cliquey_kwargs(  argument
126     def test_errors(self, device, dtype, optim_info):  argument
163         self, device, dtype, optim_info, contiguous, with_lrsched  argument
230         self, device, dtype, optim_info, with_lrsched  argument
281         self, device, dtype, optim_info  argument
318     def test_tensor_lr(self, device, dtype, optim_info):  argument
377     def test_rosenbrock_sparse(self, device, dtype, optim_info, with_lrsched):  argument
507     def test_complex(self, device, dtype, optim_info):  argument
584     def test_complex_2d(self, device, dtype, optim_info):  argument
724         optim_info,  argument
782     def test_foreach_matches_forloop(self, device, dtype, optim_info):  argument
795     def test_mixed_device_dtype(self, device, dtype, optim_info, impl):  argument
882     def test_set_default_dtype_works_with_foreach(self, device, dtype, optim_info):  argument
911     def test_foreach_large_tensor(self, device, dtype, optim_info):  argument
925     def test_peak_memory_foreach(self, device, dtype, optim_info):  argument
1020     def test_fused_matches_forloop(self, device, dtype, optim_info):  argument
1036     def test_fused_error_on_params_on_meta(self, device, dtype, optim_info):  argument
1073     def test_fused_large_tensor(self, device, dtype, optim_info):  argument
1091     def test_fused_does_not_step_if_foundinf(self, device, dtype, optim_info):  argument
1129     def test_cpu_load_state_dict(self, device, dtype, impl, optim_info):  argument
1169     def test_param_groups_weight_decay(self, device, dtype, optim_info):  argument
1211     def test_param_groups_lr(self, device, dtype, optim_info):  argument
1264     def test_step_is_noop_when_params_have_no_grad(self, device, dtype, optim_info):  argument
1283     def test_step_is_noop_for_zero_grads(self, device, dtype, optim_info):  argument
1331     def test_optimizer_can_be_printed(self, device, dtype, optim_info):  argument
1345     def test_state_dict_deterministic(self, device, dtype, optim_info):  argument
1418     def test_can_load_older_state_dict(self, device, dtype, optim_info):  argument
1466     def test_save_load_equality_with_weights_only(self, device, dtype, optim_info):  argument
1511     def test_load_nontensor_step(self, device, dtype, optim_info):  argument
1548     def test_state_dict_with_cuda_params(self, device, dtype, optim_info):  argument
1635     def test_state_dict_pre_hook(self, device, dtype, optim_info):  argument
1648     def test_state_dict_post_hook(self, device, dtype, optim_info):  argument
1661     def test_state_dict_pre_post_hook(self, device, dtype, optim_info):  argument
1699     def test_load_state_dict_pre_hook_and_prepend(self, device, dtype, optim_info):  argument
1724     def test_load_state_dict_post_hook(self, device, dtype, optim_info):  argument
1741     def test_load_state_dict_pre_post_hook(self, device, dtype, optim_info):  argument
1761     def test_step_post_hook(self, device, dtype, optim_info):  argument
1793     def test_step_pre_hook(self, device, dtype, optim_info):  argument
1825     def test_step_all_hooks(self, device, dtype, optim_info):  argument
1887     def test_deepcopy_copies_all_public_attrs(self, device, dtype, optim_info):  argument
1932         self, device, dtype, optim_info  argument
1969     def test_fused_cpu_matches_cuda(self, device, dtype, optim_info):  argument
2014     def test_defaults_changed_to_foreach(self, device, dtype, optim_info):  argument
2039     def test_non_empty_state(self, device, dtype, optim_info):  argument