checkpoints/step-2000/parity_reference_names.json

{
  "base_layer": "action_head.model.base_model.model.blocks.0.self_attn.q.weight",
  "lora": "action_head.model.base_model.model.blocks.0.self_attn.q.lora_A.default.weight",
  "action_head": "action_head.model.base_model.model.state_encoder.layer1.W"
}