`args_tensor_mask` has been removed from `torch.ops.higher_order.autograd_function_apply`

[mattteochen]

A recent Pytorch PR changed the autograd_function_apply function API by dropping the usage of args_tensor_mask keyword argument.

Currently, thunder reads this key in the kwarg dict leading to a KeyError.