Hi, this looks interesting. Could you do this for me: finding an input tensor in your model where log_softmax (or its backward, or its double backward) gives different results in 0.4.1 and 1.0.1? It would be very helpful to reproduce the issue. Thank you!