pytorch

43336f89 - [functorch] Ported `_log_softmax` to the new API (pytorch/functorch#135)

[functorch] Ported `_log_softmax` to the new API (pytorch/functorch#135) * Ported _log_softmax to the new API * Use REDUCTION_BOXED instead of VMAP_SUPPORT * Removed old code and updated the code with REDUCTION_BOXED