fix: add **kwargs to Multinomial loss' negative_gradient