https://github.com/pytorch/ao/blob/main/torchao/float8/fsdp_utils.py#L44-L48 Should it be `raise NotImplementedError("Only supports dynamic scaling")`?