| {"array_metadatas": [{"array_metadata": {"param_name": "output_projection_point.hidden_layer.kernel.value", "write_shape": [320, 1280], "chunk_shape": [320, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_projection_point.output_layer.kernel.value", "write_shape": [320, 1280], "chunk_shape": [320, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_projection_point.residual_layer.kernel.value", "write_shape": [320, 1280], "chunk_shape": [320, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_projection_quantiles.hidden_layer.kernel.value", "write_shape": [320, 1280], "chunk_shape": [320, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_projection_quantiles.output_layer.kernel.value", "write_shape": [320, 10240], "chunk_shape": [320, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_projection_quantiles.residual_layer.kernel.value", "write_shape": [320, 10240], "chunk_shape": [320, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.attn.key.kernel.value", "write_shape": [5, 1280, 16, 80], "chunk_shape": [5, 1280, 16, 80], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.attn.key_ln.scale.value", "write_shape": [5, 80], "chunk_shape": [5, 80], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.attn.out.kernel.value", "write_shape": [5, 16, 80, 1280], "chunk_shape": [5, 16, 80, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.attn.per_dim_scale.per_dim_scale.value", "write_shape": [5, 80], "chunk_shape": [5, 80], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.attn.query.kernel.value", "write_shape": [5, 1280, 16, 80], "chunk_shape": [5, 1280, 16, 80], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.attn.query_ln.scale.value", "write_shape": [5, 80], "chunk_shape": [5, 80], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.attn.value.kernel.value", "write_shape": [5, 1280, 16, 80], "chunk_shape": [5, 1280, 16, 80], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.ff0.kernel.value", "write_shape": [5, 1280, 1280], "chunk_shape": [5, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.ff1.kernel.value", "write_shape": [5, 1280, 1280], "chunk_shape": [5, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.post_attn_ln.scale.value", "write_shape": [5, 1280], "chunk_shape": [5, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.post_ff_ln.scale.value", "write_shape": [5, 1280], "chunk_shape": [5, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.pre_attn_ln.scale.value", "write_shape": [5, 1280], "chunk_shape": [5, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "stacked_xf.pre_ff_ln.scale.value", "write_shape": [5, 1280], "chunk_shape": [5, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "tokenizer.hidden_layer.bias.value", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "tokenizer.hidden_layer.kernel.value", "write_shape": [16, 1280], "chunk_shape": [16, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "tokenizer.output_layer.bias.value", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "tokenizer.output_layer.kernel.value", "write_shape": [320, 1280], "chunk_shape": [320, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "tokenizer.residual_layer.bias.value", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "tokenizer.residual_layer.kernel.value", "write_shape": [16, 1280], "chunk_shape": [16, 1280], "ext_metadata": null}}]} |