@staticmethod
def decompress(tensor, ctx):
    """Upcasts the tensor to the initialization dtype."""
    tensor_decompressed = tensor
    dtype = ctx
    if dtype.is_floating_point:
        tensor_decompressed = tensor.type(dtype)
    return tensor_decompressed
