argmax# rbc.stdlib.searching_functions.argmax(x)# Returns the indices of the maximum values along a specified axis. Array-API ‘argmax’ doc