pytorch

227d0c9c - [functorch] Fix bug in softmax/log_softmax related to scalars + refactored code to be nicer