# returncode: 0

### 0
"pkill"
"-9"
"sglang"

### 1
"sleep"
"3"

### 2
"ray"
"stop"
"--force"

### 3
"pkill"
"-9"
"ray"

### 4
"pkill"
"-9"
"python"

### 5
"sleep"
"3"

### 6
"pkill"
"-9"
"ray"

### 7
"pkill"
"-9"
"python"

### 8
"pkill"
"-9"
"redis"

### 9
"nvidia-smi"
"topo"
"-m"

### 10
"python3"
"<REPO_ROOT>/miles/utils/external_utils/model_args_utils.py"
"kimi-k2"

### 11
"ray"
"start"
"--head"
"--node-ip-address"
"10.0.0.1"
"--num-gpus"
"8"
"--disable-usage-stats"
"--dashboard-host=0.0.0.0"
"--dashboard-port=8265"

### 12
"python3"
"-c"
"import ray; ray.init(address='auto', ignore_reinit_error=True); print(int(ray.cluster_resources().get('GPU', 0))); ray.shutdown()"

### 13
"ray"
"job"
"submit"
"--address=http://127.0.0.1:8265"
"--runtime-env-json={\n  \"env_vars\": {\n    \"MC_TRANSFER_TIMEOUT\": \"300\",\n    \"RAY_DEBUG\": \"1\",\n    \"PYTHONPATH\": \"/root/Megatron-LM/\",\n    \"CUDA_DEVICE_MAX_CONNECTIONS\": \"1\",\n    \"NCCL_NVLS_ENABLE\": \"1\",\n    \"MILES_LOG_DIR\": \"\"\n  }\n}"
"--"
"python3"
"train.py"
"--disable-bias-linear"
"--num-layers"
"61"
"--hidden-size"
"7168"
"--ffn-hidden-size"
"18432"
"--num-attention-heads"
"64"
"--kv-channels"
"64"
"--normalization"
"RMSNorm"
"--position-embedding-type"
"rope"
"--rope-type"
"yarn"
"--norm-epsilon"
"1e-6"
"--swiglu"
"--untie-embeddings-and-output-weights"
"--vocab-size"
"163840"
"--multi-latent-attention"
"--q-lora-rank"
"1536"
"--kv-lora-rank"
"512"
"--qk-head-dim"
"128"
"--qk-pos-emb-head-dim"
"64"
"--v-head-dim"
"128"
"--qk-layernorm"
"--rotary-scaling-factor"
"32.0"
"--rotary-base"
"50000"
"--original-max-position-embeddings"
"4096"
"--beta-fast"
"1"
"--beta-slow"
"1"
"--mscale"
"1.0"
"--mscale-all-dim"
"1.0"
"--attention-softmax-in-fp32"
"--no-rope-fusion"
"--num-experts"
"384"
"--moe-layer-freq"
"[0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1]"
"--moe-ffn-hidden-size"
"2048"
"--moe-router-topk"
"8"
"--moe-shared-expert-intermediate-size"
"2048"
"--moe-router-pre-softmax"
"--moe-router-score-function"
"sigmoid"
"--moe-router-enable-expert-bias"
"--moe-router-load-balancing-type"
"seq_aux_loss"
"--moe-token-dispatcher-type"
"alltoall"
"--moe-aux-loss-coeff"
"0"
"--moe-router-bias-update-rate"
"0"
"--moe-router-group-topk"
"1"
"--moe-router-num-groups"
"1"
"--moe-grouped-gemm"
"--moe-router-topk-scaling-factor"
"2.827"
"--moe-router-dtype"
"fp32"
"--moe-permute-fusion"
"--hf-checkpoint"
"/root/models/Kimi-K2-Instruct/"
"--ref-load"
"/root/multinode/Kimi-K2-Instruct_torch_dist/"
"--prompt-data"
"/root/datasets/dapo-math-17k/dapo-math-17k.jsonl"
"--input-key"
"prompt"
"--label-key"
"label"
"--apply-chat-template"
"--rollout-shuffle"
"--rm-type"
"deepscaler"
"--num-rollout"
"13"
"--rollout-batch-size"
"8"
"--n-samples-per-prompt"
"8"
"--rollout-max-response-len"
"100"
"--rollout-temperature"
"1"
"--global-batch-size"
"64"
"--balance-data"
"--eval-prompt-data"
"aime"
"/root/datasets/aime-2024/aime-2024.jsonl"
"--n-samples-per-eval-prompt"
"16"
"--eval-max-response-len"
"16384"
"--eval-top-p"
"0.7"
"--optimizer"
"adam"
"--lr"
"1e-6"
"--lr-decay-style"
"constant"
"--weight-decay"
"0.1"
"--adam-beta1"
"0.9"
"--adam-beta2"
"0.98"
"--optimizer-cpu-offload"
"--overlap-cpu-optimizer-d2h-h2d"
"--use-precision-aware-optimizer"
"--advantage-estimator"
"grpo"
"--use-kl-loss"
"--kl-loss-coef"
"0.00"
"--kl-loss-type"
"low_var_kl"
"--entropy-coef"
"0.00"
"--eps-clip"
"0.2"
"--eps-clip-high"
"0.28"
"--tensor-model-parallel-size"
"8"
"--sequence-parallel"
"--pipeline-model-parallel-size"
"8"
"--decoder-last-pipeline-num-layers"
"5"
"--context-parallel-size"
"4"
"--expert-model-parallel-size"
"32"
"--expert-tensor-parallel-size"
"1"
"--recompute-granularity"
"full"
"--recompute-method"
"uniform"
"--recompute-num-layers"
"1"
"--use-dynamic-batch-size"
"--max-tokens-per-gpu"
"16384"
"--rollout-num-gpus-per-engine"
"32"
"--rollout-num-gpus"
"256"
"--sglang-mem-fraction-static"
"0.7"
"--sglang-ep-size"
"32"
"--sglang-dp-size"
"8"
"--sglang-moe-dense-tp-size"
"1"
"--sglang-enable-dp-attention"
"--sglang-enable-dp-lm-head"
"--sglang-server-concurrency"
"1024"
"--sglang-cuda-graph-bs-decode"
"1"
"2"
"4"
"8"
"16"
"--sglang-moe-runner-backend"
"triton"
"--sglang-fp8-gemm-backend"
"triton"
"--sglang-remote-instance-weight-loader-start-seed-via-transfer-engine"
"--sglang-remote-instance-weight-loader-ib-device"
"{\"0\":\"ibp0\",\"1\":\"ibp1\",\"2\":\"ibp2\",\"3\":\"ibp3\",\"4\":\"ibp4\",\"5\":\"ibp5\",\"6\":\"ibp6\",\"7\":\"ibp7\"}"
"--sglang-model-loader-extra-config"
"{\"enable_multithread_load\":true,\"num_threads\":8}"
"--attention-dropout"
"0.0"
"--hidden-dropout"
"0.0"
"--accumulate-allreduce-grads-in-fp32"
"--attention-softmax-in-fp32"
"--attention-backend"
"flash"
"--moe-token-dispatcher-type"
"alltoall"
"--actor-num-nodes"
"32"
"--actor-num-gpus-per-node"
"8"
"--update-weight-buffer-size"
"2147483648"
"--check-weight-update-equal"
"--update-weight-transfer-mode"
"p2p"
