Hi,
You will most likely have to either do a for loop over all the parameters to find the one with largest value.
If you actually need a topk, it might be simplet to concatenate all the grads in a single Tensor and then call topk on that Tensor.