pytorch

7f344c5a - [functorch] Add batch rule for nll_loss_forward for most common cases