PyTorch
API: torch.get_default_device
Strategy: Macro 'torch.device('cuda' if torch.cuda.is_available() else 'cpu')'
JAX (Core)
API: jax.default_backend
Strategy: Macro 'jax.default_backend()'
Apple MLX
API: mlx.core.default_device
Strategy: Direct Mapping
Flax NNX
API: jax.default_backend
Strategy: Macro 'jax.default_backend()'
PaxML / Praxis
API: —
Strategy: Macro 'jax.default_backend()'