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
  cpu: 2
