{"array_metadatas": [{"array_metadata": {"param_name": "params.critic.params.q_head.linear.bias.value", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_head.linear.kernel.value", "write_shape": [32, 1], "chunk_shape": [32, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 288, 269], "chunk_shape": [27, 288, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 288], "chunk_shape": [27, 269, 288], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 18, 16, 72], "chunk_shape": [27, 18, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 4, 72, 72], "chunk_shape": [27, 4, 72, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 18, 16, 72], "chunk_shape": [27, 18, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 18, 16, 72], "chunk_shape": [27, 18, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.embedding.bias.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 18], "chunk_shape": [14, 14, 3, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.head.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.head.kernel.value", "write_shape": [288, 128], "chunk_shape": [288, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.img.pos_embedding.value", "write_shape": [1, 4, 1152], "chunk_shape": [1, 4, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [4018, 2048], "chunk_shape": [4018, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.final_norm.scale.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 2, 256, 128], "chunk_shape": [18, 2, 256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 32, 256], "chunk_shape": [18, 2, 1, 32, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 2, 128, 256], "chunk_shape": [18, 2, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 512, 1024], "chunk_shape": [18, 2, 512, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 256, 2048], "chunk_shape": [18, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 32], "chunk_shape": [18, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 32], "chunk_shape": [18, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.action_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.action_proj.kernel.value", "write_shape": [14, 32], "chunk_shape": [14, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.cls_token.value", "write_shape": [1, 1, 32], "chunk_shape": [1, 1, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.state_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.q_network.state_proj.kernel.value", "write_shape": [14, 32], "chunk_shape": [14, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_head.linear.bias.value", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_head.linear.kernel.value", "write_shape": [32, 1], "chunk_shape": [32, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 288, 269], "chunk_shape": [27, 288, 269], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 269, 288], "chunk_shape": [27, 269, 288], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 18, 16, 72], "chunk_shape": [27, 18, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 18], "chunk_shape": [27, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 4, 72, 72], "chunk_shape": [27, 4, 72, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 18, 16, 72], "chunk_shape": [27, 18, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 18, 16, 72], "chunk_shape": [27, 18, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.embedding.bias.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 18], "chunk_shape": [14, 14, 3, 18], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.head.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.head.kernel.value", "write_shape": [288, 128], "chunk_shape": [288, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.img.pos_embedding.value", "write_shape": [1, 4, 1152], "chunk_shape": [1, 4, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [4018, 2048], "chunk_shape": [4018, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.final_norm.scale.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 2, 256, 128], "chunk_shape": [18, 2, 256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 32, 256], "chunk_shape": [18, 2, 1, 32, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 2, 128, 256], "chunk_shape": [18, 2, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 512, 1024], "chunk_shape": [18, 2, 512, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 256, 2048], "chunk_shape": [18, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 32], "chunk_shape": [18, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 32], "chunk_shape": [18, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.action_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.action_proj.kernel.value", "write_shape": [14, 32], "chunk_shape": [14, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.cls_token.value", "write_shape": [1, 1, 32], "chunk_shape": [1, 1, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.state_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic.params.target_q_network.state_proj.kernel.value", "write_shape": [14, 32], "chunk_shape": [14, 32], "ext_metadata": null}}]}