{"record":{"id":"49401833eccf8a99","repo":"keras-team/keras","slug":"device-not-found-device-name","errorCode":null,"errorMessage":"Device not found: {device_name}","messagePattern":"Device not found: (.+?)","errorType":"validation","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"keras/src/backend/jax/distribution_lib.py","lineNumber":188,"sourceCode":"def process_id():\n    \"\"\"Return the current process ID for the distribution setting.\"\"\"\n    return jax.process_index()\n\n\ndef _to_backend_device(device_name):\n    if isinstance(device_name, jax.Device):\n        return device_name\n    device_name = str(device_name)\n    if \":\" not in device_name:\n        device_type, device_id = device_name, 0\n    else:\n        device_type, device_id = device_name.split(\":\")\n\n    devices = jax.devices(backend=device_type)\n    for device in devices:\n        if device.platform == device_type and device.id == int(device_id):\n            return device\n    raise ValueError(f\"Device not found: {device_name}\")\n\n\ndef _to_backend_mesh(device_mesh):\n    \"\"\"Convert the DeviceMesh to JAX backend specific Mesh.\n\n    Args:\n        device_mesh: DeviceMesh instance to convert.\n\n    Returns:\n        A `jax.sharding.Mesh` instance.\n    \"\"\"\n    shape = device_mesh.devices.shape\n    devices = [_to_backend_device(d) for d in device_mesh.devices.flatten()]\n    devices = np.array(devices).reshape(shape)\n    return jax.sharding.Mesh(devices, device_mesh.axis_names)\n\n\ndef _to_backend_layout(tensor_layout):","sourceCodeStart":170,"sourceCodeEnd":206,"githubUrl":"https://github.com/keras-team/keras/blob/7a34a03db60bf60042242d6a556fc3be119046a5/keras/src/backend/jax/distribution_lib.py#L170-L206","documentation":"Error \"Device not found: {device_name}\" thrown in keras-team/keras.","triggerScenarios":"Thrown at keras/src/backend/jax/distribution_lib.py:188 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"7a34a03db60bf60042242d6a556fc3be119046a5","analyzedAt":"2026-08-25T21:25:25.994Z","schemaVersion":2},"datasetVersion":"2026-08-26T02:17:13.382Z"}