chain_name: "Aura G chain"
description: "Aura G chain, 9 steps."

# Global settings
settings:
  base_dir: "/home/tony/Work/tony/slurm/bluejay"
  log_dir: "/home/tony/slurm/logs"
  
  # SLURM defaults - no partition specified
  slurm_defaults:
    time: "48:00:00"
    ntasks_per_node: 8
    cpus_per_task: 4
    gpus_per_node: 8

# Job definitions
jobs:
  # - name: "round_1"
  #   script: "run_ipo_bluejay_r1.sh"
  #   nodes: 8
  #   model_cache_loss_name: "6b_bjay_29_0"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r1"

  # - name: "round_2"
  #   script: "run_ipo_bluejay_r2_task.sh"
  #   nodes: 8
  #   model_cache_loss_name: "6b_bjay_29_1"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r2"
    
  # - name: "round_3"
  #   script: "run_ipo_bluejay_r3_exp.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_29_2"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r3"

  # - name: "round_4"
  #   script: "run_ipo_bluejay_r4_mix.sh"
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_29_3"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r4"
    
  # - name: "round_5"
  #   script: "run_ipo_bluejay_r5_mix_2.sh"  
  #   nodes: 10
  #   model_cache_loss_name: "6b_bjay_29_4"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r5"

  # - name: "round_6"
  #   script: "run_ipo_bluejay_r6_auk.sh"
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_29_5"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r6"
    
  # - name: "round_7"
  #   script: "run_ipo_bluejay_r7_auk_2.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_29_6"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r7"
    
  # - name: "round_8"
  #   script: "run_ipo_bluejay_r8_auk_3.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_29_7"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r8"
    
  # - name: "round_9"
  #   script: "run_ipo_bluejay_r9_auk_4.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_29_8"
  #   wandb_run_name: "dpo_6b_base0519_bjay_29_r9"
    
  # - name: "round_10"
  #   script: "run_ipo_bluejay_r10_bluejay_1.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_20_r13_v25"
  #   wandb_run_name: "dpo_6b_base0519_bjay_20_r14_v24"
    
  # - name: "round_11"
  #   script: "run_ipo_bluejay_r11_bluejay_2.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_20_r14_v24"
  #   wandb_run_name: "dpo_6b_base0519_bjay_20_r15_v25"

  # - name: "round_12"
  #   script: "run_ipo_bluejay_r12_bluejay_3.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_27_d43_6"
  #   wandb_run_name: "dpo_6b_base0519_bjay_27_d43_r7_v26"
    
  # - name: "round_13"
  #   script: "run_ipo_bluejay_r13_bluejay_4.sh"  
  #   nodes: 16
  #   model_cache_loss_name: "6b_bjay_27_d43_7"
  #   wandb_run_name: "dpo_6b_base0519_bjay_27_d43_r8_v27"