def rank(
        self, tokens: List[List[mx.array]], sum_logprobs: List[List[float]]
    ) -> List[int]:
        """
        Given a list of groups of samples and their cumulative log probabilities,
        return the indices of the samples in each group to select as the final result
        """
        raise NotImplementedError

        Given a list of groups of samples and their cumulative log probabilities,
        return the indices of the samples in each group to select as the final result
        
mlx_audio/stt/models/whisper/decoding.py