keras/src/distribution/distribution_lib.py/DeviceMesh/__repr__
class DeviceMesh:    def __repr__(self):
        return (
            f"<{self.__class__.__name__} "
            f"shape={self.shape}, axis_names={self.axis_names}>"
        )