xla

Rectify test_zero1.py once optim.load_state_dict doesn't guarantee immutability

#5382

Merged