import torch from importlib.util import find_spec def amp_definitely_not_available(): return not (torch.cuda.is_available() or find_spec('torch_xla'))