torch.optim — PyTorch 2.7 documentation