{"array_metadatas": [{"array_metadata": {"param_name": "embed_tokens.embedding.value", "write_shape": [50257, 64], "chunk_shape": [50257, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.input_layernorm.weight.value", "write_shape": [1, 512], "chunk_shape": [1, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.mlp.down_proj.kernel.value", "write_shape": [1, 2048, 512], "chunk_shape": [1, 2048, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.mlp.gate_proj.kernel.value", "write_shape": [8, 64, 2048], "chunk_shape": [8, 64, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.mlp.up_proj.kernel.value", "write_shape": [8, 64, 2048], "chunk_shape": [8, 64, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.post_attention_layernorm.weight.value", "write_shape": [1, 512], "chunk_shape": [1, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.self_attn.k_proj.kernel.value", "write_shape": [8, 64, 512], "chunk_shape": [8, 64, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.self_attn.o_proj.kernel.value", "write_shape": [1, 512, 512], "chunk_shape": [1, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.self_attn.q_proj.kernel.value", "write_shape": [8, 64, 512], "chunk_shape": [8, 64, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "layers.self_attn.v_proj.kernel.value", "write_shape": [8, 64, 512], "chunk_shape": [8, 64, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "lm_head.kernel.value", "write_shape": [64, 50257], "chunk_shape": [64, 50257], "ext_metadata": null}}, {"array_metadata": {"param_name": "norm.weight.value", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}]}