| {"array_metadatas": [{"array_metadata": {"param_name": "opt_state.notfinite_count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.last_finite", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.total_notfinite", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block0_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block0_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block0_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block0_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block0_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block0_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block0_proj.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block1_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block1_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block1_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block1_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block1_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block1_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block2_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block2_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block2_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block2_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block2_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block2_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block3_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block3_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block3_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block3_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block3_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block3_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block4_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block4_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block4_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block4_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block4_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block4_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block5_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block5_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block5_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block5_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block5_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block5_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block6_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block6_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block6_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block6_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block6_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block6_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block7_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block7_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block7_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block7_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block7_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.block7_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.output.bias", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_0", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_1", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_2", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_3", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_4", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_5", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_6", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor.res_gate_7", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block0_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block0_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block0_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block0_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block0_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block0_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block0_proj.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block1_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block1_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block1_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block1_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block1_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block1_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block2_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block2_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block2_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block2_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block2_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block2_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block3_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block3_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block3_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block3_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block3_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block3_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block4_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block4_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block4_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block4_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block4_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block4_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block5_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block5_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block5_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block5_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block5_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block5_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block6_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block6_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block6_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block6_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block6_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block6_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block7_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block7_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block7_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block7_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block7_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.block7_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.output.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_0", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_1", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_2", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_3", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_4", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_5", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_6", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic.res_gate_7", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.log_std", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block0_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block0_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block0_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block0_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block0_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block0_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block0_proj.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block1_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block1_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block1_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block1_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block1_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block1_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block2_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block2_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block2_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block2_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block2_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block2_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block3_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block3_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block3_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block3_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block3_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block3_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block4_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block4_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block4_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block4_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block4_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block4_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block5_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block5_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block5_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block5_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block5_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block5_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block6_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block6_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block6_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block6_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block6_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block6_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block7_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block7_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block7_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block7_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block7_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.block7_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.output.bias", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_0", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_1", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_2", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_3", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_4", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_5", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_6", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor.res_gate_7", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block0_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block0_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block0_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block0_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block0_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block0_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block0_proj.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block1_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block1_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block1_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block1_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block1_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block1_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block2_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block2_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block2_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block2_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block2_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block2_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block3_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block3_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block3_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block3_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block3_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block3_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block4_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block4_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block4_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block4_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block4_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block4_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block5_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block5_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block5_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block5_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block5_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block5_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block6_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block6_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block6_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block6_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block6_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block6_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block7_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block7_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block7_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block7_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block7_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.block7_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.output.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_0", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_1", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_2", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_3", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_4", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_5", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_6", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic.res_gate_7", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.log_std", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block0_layer0_dense.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block0_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block0_proj.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block1_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block1_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block2_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block2_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block3_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block3_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block4_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block4_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block5_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block5_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block6_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block6_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block7_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.block7_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor.output.kernel", "write_shape": [1024, 354], "chunk_shape": [1024, 354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block0_layer0_dense.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block0_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block0_proj.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block1_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block1_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block2_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block2_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block3_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block3_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block4_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block4_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block5_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block5_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block6_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block6_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block7_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.block7_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic.output.kernel", "write_shape": [1024, 1], "chunk_shape": [1024, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.ns_coeffs", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer0_dense.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_proj.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block0_proj.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block1_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block2_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block3_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block4_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block5_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block6_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.block7_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.output.bias", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.output.kernel", "write_shape": [1024, 354], "chunk_shape": [1024, 354], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_0", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_1", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_2", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_3", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_4", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_5", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_6", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor.res_gate_7", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer0_dense.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_proj.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block0_proj.kernel", "write_shape": [1710, 1024], "chunk_shape": [1710, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block1_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block2_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block3_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block4_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block5_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block6_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer0_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer0_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer0_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer0_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer1_dense.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer1_dense.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer1_ln.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.block7_layer1_ln.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.output.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.output.kernel", "write_shape": [1024, 1], "chunk_shape": [1024, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_0", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_1", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_2", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_3", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_4", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_5", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_6", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.res_gate_7", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.log_std", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "run_stats.RunningMeanStd_0.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "run_stats.RunningMeanStd_0.mean", "write_shape": [1710], "chunk_shape": [1710], "ext_metadata": null}}, {"array_metadata": {"param_name": "run_stats.RunningMeanStd_0.var", "write_shape": [1710], "chunk_shape": [1710], "ext_metadata": null}}]} |