Files
mlx/mlx/backend/cuda/arg_reduce.cu