Slot: accelerator

Hardware accelerator the run targets. ‘auto’ (default) lets JAX detect the machine — GPU/TPU when present, else CPU; ‘cpu’, ‘gpu’, or ‘tpu’ pin the JAX platform explicitly. tvbo maps this to JAX_PLATFORMS before JAX initialises (an existing JAX_PLATFORMS in the environment still wins); tvboptim then spreads work across that platform’s devices via n_pmap/n_vmap. In a workflow kit, ‘gpu’ also makes the experiment’s rule request a SLURM gres unless one is set.

URI: tvbo:slot/accelerator

Applicable Classes

Name Description Modifies Slot
ExecutionConfig Configuration for computational execution (parallelization, precision, hardwa… no

Properties

Type and Range

Property Value
Range xsd:string
Domain Of ExecutionConfig

Cardinality and Requirements

Property Value

Slot Characteristics

Property Value
If Absent string(auto)
Owner ExecutionConfig

Identifier and Mapping Information

Schema Source

  • from schema: https://w3id.org/tvbo

Mappings

Mapping Type Mapped Value
self tvbo:accelerator
native tvbo:accelerator

LinkML Source

name: accelerator
description: Hardware accelerator the run targets. 'auto' (default) lets JAX detect
  the machine — GPU/TPU when present, else CPU; 'cpu', 'gpu', or 'tpu' pin the JAX
  platform explicitly. tvbo maps this to JAX_PLATFORMS before JAX initialises (an
  existing JAX_PLATFORMS in the environment still wins); tvboptim then spreads work
  across that platform's devices via n_pmap/n_vmap. In a workflow kit, 'gpu' also
  makes the experiment's rule request a SLURM gres unless one is set.
from_schema: https://w3id.org/tvbo
rank: 1000
ifabsent: string(auto)
owner: ExecutionConfig
domain_of:
- ExecutionConfig
range: string