def step(self, net: NeuralNet) -> None: for param, grad in net.params_and_grad(): param -= self.lr * grad