code: def expand_platform_alias(platform: str) -> list[str]:
      """Expands, e.g., "gpu" to ["cuda", "rocm"].
  
      This is used for convenience reasons: we expect cuda and rocm to act similarly
      in many respects since they share most of the same code.
      """
      return _alias_to_platforms.get(platform, [platform])
  
imports: from __future__ import annotations
  from __future__ import annotations
  from collections.abc import Mapping
  import dataclasses
  from functools import lru_cache, partial
  import importlib
  import json
  import logging
  import os
  import pkgutil
  import platform
  import sys
  import threading
  from typing import Any, Callable, Union
  import warnings
  from jax._src import config
  from jax._src import distributed
  from jax._src import traceback_util
  from jax._src import util
  from jax._src import hardware_utils
  from jax._src.cloud_tpu_init import maybe_import_libtpu
  from jax._src.lib import cuda_versions
  from jax._src.lib import xla_client
  from jax._src.lib import xla_extension
  from jax._src.lib import xla_extension_version
label_desc: Use image node_id 15 for calling a global function with example usage: expand_platform_alias(platform) and returns: _alias_to_platforms