CudaAvailable

Auto-generated from jax_code_defs

PyTorchJAX (Core)TensorFlow

PyTorch

API: torch.cuda.is_available

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.devices

Strategy: Plugin (cuda_is_available)

TensorFlow

API: tf.config.list_physical_devices

Strategy: Plugin (cuda_is_available)