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