src/transformers/models/maskformer/modeling_maskformer.py/MaskFormerLoss/_max_by_axis
class MaskFormerLoss:    def _max_by_axis(self, the_list: List[List[int]]) -> List[int]:
        maxes = the_list[0]
        for sublist in the_list[1:]:
            for index, item in enumerate(sublist):
                maxes[index] = max(maxes[index], item)
        return maxes