train_state/array_metadatas/process_0

{"array_metadatas": [{"array_metadata": {"param_name": "opt_state.notfinite_count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.last_finite", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.total_notfinite", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_dense_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_dense_ln_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_dense_ln_0.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_output.bias", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_rnn.cell.hn.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_rnn.cell.in.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_rnn.cell.ir.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_rnn.cell.iz.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_rnn_input.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_rnn_input_ln.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.actor_rnn_input_ln.scale", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_1.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_10.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_11.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_12.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_13.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_14.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_15.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_2.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_3.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_4.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_5.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_6.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_7.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_8.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_9.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_0.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_1.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_1.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_10.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_10.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_11.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_11.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_12.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_12.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_13.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_13.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_14.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_14.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_15.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_15.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_2.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_2.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_3.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_3.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_4.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_4.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_5.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_5.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_6.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_6.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_7.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_7.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_8.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_8.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_9.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_dense_ln_9.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.critic_output.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.mu.log_std", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_dense_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_dense_ln_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_dense_ln_0.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_output.bias", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_rnn.cell.hn.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_rnn.cell.in.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_rnn.cell.ir.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_rnn.cell.iz.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_rnn_input.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_rnn_input_ln.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.actor_rnn_input_ln.scale", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_1.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_10.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_11.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_12.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_13.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_14.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_15.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_2.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_3.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_4.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_5.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_6.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_7.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_8.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_9.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_0.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_1.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_1.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_10.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_10.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_11.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_11.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_12.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_12.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_13.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_13.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_14.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_14.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_15.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_15.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_2.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_2.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_3.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_3.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_4.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_4.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_5.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_5.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_6.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_6.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_7.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_7.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_8.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_8.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_9.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_dense_ln_9.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.critic_output.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.0.nu.log_std", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.adam.inner_state.2.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_dense_0.kernel", "write_shape": [1840, 1024], "chunk_shape": [1840, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_output.kernel", "write_shape": [1024, 354], "chunk_shape": [1024, 354], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_rnn.cell.hn.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_rnn.cell.hr.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_rnn.cell.hz.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_rnn.cell.in.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_rnn.cell.ir.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_rnn.cell.iz.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.actor_rnn_input.kernel", "write_shape": [2418, 1840], "chunk_shape": [2418, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_0.kernel", "write_shape": [2418, 1024], "chunk_shape": [2418, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_1.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_10.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_11.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_12.kernel", "write_shape": [1024, 2048], "chunk_shape": [1024, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_13.kernel", "write_shape": [2048, 2048], "chunk_shape": [2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_14.kernel", "write_shape": [2048, 2048], "chunk_shape": [2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_15.kernel", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_2.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_3.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_4.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_5.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_6.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_7.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_8.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_dense_9.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.mu.critic_output.kernel", "write_shape": [1024, 1], "chunk_shape": [1024, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.0.ns_coeffs", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "opt_state.inner_state.1.inner_states.muon.inner_state.3.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_dense_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_dense_0.kernel", "write_shape": [1840, 1024], "chunk_shape": [1840, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_dense_ln_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_dense_ln_0.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_output.bias", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_output.kernel", "write_shape": [1024, 354], "chunk_shape": [1024, 354], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.hn.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.hn.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.hr.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.hz.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.in.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.in.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.ir.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.ir.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.iz.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn.cell.iz.kernel", "write_shape": [1840, 1840], "chunk_shape": [1840, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn_input.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn_input.kernel", "write_shape": [2418, 1840], "chunk_shape": [2418, 1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn_input_ln.bias", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.actor_rnn_input_ln.scale", "write_shape": [1840], "chunk_shape": [1840], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_0.kernel", "write_shape": [2418, 1024], "chunk_shape": [2418, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_1.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_1.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_10.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_10.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_11.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_11.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_12.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_12.kernel", "write_shape": [1024, 2048], "chunk_shape": [1024, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_13.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_13.kernel", "write_shape": [2048, 2048], "chunk_shape": [2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_14.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_14.kernel", "write_shape": [2048, 2048], "chunk_shape": [2048, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_15.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_15.kernel", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_2.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_2.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_3.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_3.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_4.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_4.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_5.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_5.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_6.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_6.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_7.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_7.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_8.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_8.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_9.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_9.kernel", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_0.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_0.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_1.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_1.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_10.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_10.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_11.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_11.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_12.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_12.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_13.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_13.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_14.bias", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_14.scale", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_15.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_15.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_2.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_2.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_3.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_3.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_4.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_4.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_5.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_5.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_6.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_6.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_7.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_7.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_8.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_8.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_9.bias", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_dense_ln_9.scale", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_output.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.critic_output.kernel", "write_shape": [1024, 1], "chunk_shape": [1024, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.log_std", "write_shape": [354], "chunk_shape": [354], "ext_metadata": null}}, {"array_metadata": {"param_name": "run_stats.RunningMeanStd_0.count", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "run_stats.RunningMeanStd_0.mean", "write_shape": [2418], "chunk_shape": [2418], "ext_metadata": null}}, {"array_metadata": {"param_name": "run_stats.RunningMeanStd_0.var", "write_shape": [2418], "chunk_shape": [2418], "ext_metadata": null}}]}