xma.optimizers.sgd.op

sgd(parameters: list[Tensor], gradients: list[Tensor], lr: float, maximize: bool, horizontal_fusion: bool, weight_decay: float, momentum: float, dampening: float, nesterov: bool, *, kernel_backend: KernelBackend | None = None) → None