Thanks ptr. Is there a list of functions like this which “almost always” break the gradient or do we just have to experiment?
Using min() and max() most likely have similar behavior as round() but does not throw “gradient broken error” like argmin() and argmax() do.