{"array_metadatas": [{"array_metadata": {"param_name": "dropout.rngs.count.value", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "dropout.rngs.key.value", "write_shape": [2], "chunk_shape": [2], "ext_metadata": {"random_key_impl": "threefry2x32"}}}, {"array_metadata": {"param_name": "input_proj.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "input_proj.kernel.value", "write_shape": [16, 256], "chunk_shape": [16, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.embedder.embedding.value", "write_shape": [1, 256], "chunk_shape": [1, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.final_norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.A_log.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.D.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.conv1d.conv.bias.value", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.conv1d.conv.kernel.value", "write_shape": [4, 1, 768], "chunk_shape": [4, 1, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.dt_bias.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.in_proj.kernel.value", "write_shape": [256, 1288], "chunk_shape": [256, 1288], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.norm.weight.value", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.out_proj.kernel.value", "write_shape": [512, 256], "chunk_shape": [512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.A_log.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.D.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.conv1d.conv.bias.value", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.conv1d.conv.kernel.value", "write_shape": [4, 1, 768], "chunk_shape": [4, 1, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.dt_bias.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.in_proj.kernel.value", "write_shape": [256, 1288], "chunk_shape": [256, 1288], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.norm.weight.value", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.out_proj.kernel.value", "write_shape": [512, 256], "chunk_shape": [512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_proj.bias.value", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_proj.kernel.value", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}]}