chief: &gpu
mem: 1.4Ti
cpu: 24
num_accelerators: 16
accelerator_type: a100
dataset_dispatcher:
mem: 2Gi
cpu: 2
num_dataset_workers: 4
dataset_worker:
mem: 14Gi