OptionalcommandWorkspace path of the script to run on each node in this deployment. Upload the script to this path and supply the path here. When the task runs, the file at this path is run on each node; if it fails, the task fails with its exit code.
Example script contents:
python train.py --epochs 10
accelerate launch train.py --config config.yaml
torchrun --nproc_per_node=8 train.py
OptionalcomputeCompute resources allocated to each node in this deployment.
OptionalnameOptional human-readable name for this deployment (for example, driver,
worker, param_server). Used for log and UI display. Distinct names
are recommended so deployments can be told apart, but uniqueness is not
enforced.
DeploymentSpec: configuration for one deployment within an AiRuntimeTask. Each entry in
AiRuntimeTask.deploymentsdescribes a group of nodes that share the same command and compute. Many single-program training algorithms use a single entry where every node runs the same command; role-split workloads (driver + worker, parameter server, separate eval node, etc.) use multiple entries.