def METHOD_NAME(self) -> None:
    model = BasicModel_MultiLayer(multi_input_module=True)
    inp = (torch.randn(6, 3), torch.randn(6, 3))
    self._compare_sample_grads_per_sample(model, inp, lambda x: torch.sum(x), "sum")