| {"array_metadatas": [{"array_metadata": {"param_name": "decoder_layers.mlp.latent_mlp.odot", "write_shape": [16, 2048, 2048], "chunk_shape": [16, 2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_noncausal.self_attn.k_mixer.conv.kernel", "write_shape": [6, 5, 1, 2048], "chunk_shape": [6, 5, 1, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_noncausal.self_attn.k_proj.kernel", "write_shape": [6, 2048, 2048], "chunk_shape": [6, 2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_noncausal.self_attn.q_mixer.conv.kernel", "write_shape": [6, 5, 1, 2048], "chunk_shape": [6, 5, 1, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_noncausal.self_attn.v_mixer.conv.kernel", "write_shape": [6, 5, 1, 2048], "chunk_shape": [6, 5, 1, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_noncausal.self_attn.v_proj.kernel", "write_shape": [6, 2048, 2048], "chunk_shape": [6, 2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_noncausal_norm.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_state_scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_state_shift", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_writer.key_proj.kernel", "write_shape": [2048, 2048], "chunk_shape": [2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_writer.value_gate_proj.kernel", "write_shape": [2048, 2048], "chunk_shape": [2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_writer.value_proj.kernel", "write_shape": [2048, 2048], "chunk_shape": [2048, 2048], "ext_metadata": null}}]} |