keras/src/utils/torch_utils.py/TorchModuleWrapper/get_config
class TorchModuleWrapper:    def get_config(self):
        base_config = super().get_config()
        import torch

        buffer = io.BytesIO()
        torch.save(self.module, buffer)
        config = {"module": buffer.getvalue()}
        return {**base_config, **config}