pytorch

9a6b91b6 - [functorch] Port jvp decompositions to unify on python one (pytorch/functorch#772)

[functorch] Port jvp decompositions to unify on python one (pytorch/functorch#772) * Port nll_loss_backward decomposition to unify on python one * made some changes * clean up unneded parts now * add log_softmax and softmax decomps