The whole point of using a Function is to not use the autograd.
If you want to use the autograd, you can just use a regular python function.
If you don’t return gradients for one of the inputs, you can return None for it.