  elif backend == "apple_silicon":
      from .apple_silicon import AppleSiliconStats as GPUStats  # type:ignore
  elif backend == "directml":
      from .directml import DirectML as GPUStats  # type:ignore
  elif backend == "cpu":
      from .cpu import CPUStats as GPUStats  # type:ignore
 
</s> remove         self._env.update_tf_dep()
 </s> add         self._check_rocm() </s> remove ValidBackends = Literal["amd", "nvidia", "cpu", "apple_silicon", "directml"]
 </s> add ValidBackends = Literal["amd", "nvidia", "cpu", "apple_silicon", "directml", "rocm"]
elif backend == "rocm":
    from .rocm import ROCm as GPUStats  # type:ignore
else:
  elif backend == "apple_silicon":
      from .apple_silicon import AppleSiliconStats as GPUStats  # type:ignore
  elif backend == "directml":
      from .directml import DirectML as GPUStats  # type:ignore
elif backend == "rocm":
    from .rocm import ROCm as GPUStats  # type:ignore
else:
      from .cpu import CPUStats as GPUStats  # type:ignore
keep keep keep keep replace keep
 Linux - ROCm (AMD) support 
lib/gpu_stats/__init__.py