Is there any work around to make a differentiable torch.max?

autograd
2

Please see this issue