{"array_metadatas": [{"array_metadata": {"param_name": "params.action_embedding.action_projection_bias", "write_shape": [2, 768], "chunk_shape": [2, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_embedding.action_projection_kernel", "write_shape": [2, 14, 768], "chunk_shape": [2, 14, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_embedding.base_token", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.observation_projection.bias", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.observation_projection.kernel", "write_shape": [16, 768], "chunk_shape": [16, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.prediction_projection.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.prediction_projection.kernel", "write_shape": [768, 16], "chunk_shape": [768, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.register_tokens", "write_shape": [2, 768], "chunk_shape": [2, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.shortcut_embedding.signal_embedding.embedding", "write_shape": [129, 384], "chunk_shape": [129, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.shortcut_embedding.step_embedding.embedding", "write_shape": [8, 384], "chunk_shape": [8, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_0.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_1.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_2.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_3.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_4.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_5.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_6.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.Attention_0.Dense_0.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.Attention_0.Dense_1.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.Attention_0.Dense_2.kernel", "write_shape": [768, 256], "chunk_shape": [768, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.Attention_0.Dense_3.kernel", "write_shape": [768, 768], "chunk_shape": [768, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.Attention_0.RMSNorm_0.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.Attention_0.RMSNorm_1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.RMSNorm_0.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.RMSNorm_1.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.SwiGLU_0.Dense_0.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.SwiGLU_0.Dense_1.kernel", "write_shape": [768, 2304], "chunk_shape": [768, 2304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.TransformerBlock_7.SwiGLU_0.Dense_2.kernel", "write_shape": [2304, 768], "chunk_shape": [2304, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.transformer.final_norm.scale", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}]}