diff --git a/.gitattributes b/.gitattributes
index c9828655d9c4426ed042aa0467d83911a3377537..694f0af62480bf34e5990d6dae3ec3f7979f3213 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -380,3 +380,8 @@ InvertedDoublePendulum/AE_MAD_SAC_kappa0p05_maxalpha6_gpu1h/utd2/10seed/checkpoi
 Ant/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_55/checkpoint/2300000/default/ocdbt.process_0/d/91f99ce5e82c9485b9449e3ba09fe136 filter=lfs diff=lfs merge=lfs -text
 Ant/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_55/checkpoint/2300000/default/ocdbt.process_0/d/c02ba92769bc7fc0b693b2c3ae5a9594 filter=lfs diff=lfs merge=lfs -text
 Ant/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_55/checkpoint/2300000/default/ocdbt.process_0/d/cd3e0d3dc730cd4af0e1d31ce1d969e6 filter=lfs diff=lfs merge=lfs -text
+Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/4385a586b5625eeb416b5c6a8c4aaad6 filter=lfs diff=lfs merge=lfs -text
+Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/85dfee5343855c9cde0310dab8671e60 filter=lfs diff=lfs merge=lfs -text
+Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/9bf96da929a70ea41e16bd0d957238c3 filter=lfs diff=lfs merge=lfs -text
+Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/b25c2843d529ba5ba9bfa8dfd66082ad filter=lfs diff=lfs merge=lfs -text
+Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/df28d728678e56ae0549ea3bcef7da69 filter=lfs diff=lfs merge=lfs -text
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/_CHECKPOINT_METADATA b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fa1a7642a347c5af84c948d3cfd9a2dc465113f9
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1778251004269125424, "commit_timestamp_nsecs": 1778251006856626514, "custom_metadata": {}}
\ No newline at end of file
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/_METADATA b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f9917614c83b158fd425a25615b61b1e1ca0c4fd
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('critic', 'step')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'params', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '0')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('critic', 'opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'mu', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '0', 'nu', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic', 'opt_state', '1', '1')": {"key_metadata": [{"key": "critic", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('critic_target', 'step')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'params', 'params', 'BinnedRegression_1', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "BinnedRegression_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('critic_target', 'opt_state')": {"key_metadata": [{"key": "critic_target", "key_type": 2}, {"key": "opt_state", "key_type": 2}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('actor', 'step')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'params', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '0')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('actor', 'opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('actor', 'opt_state', '1', '1')": {"key_metadata": [{"key": "actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('encoder', 'step')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'params', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '0')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('encoder', 'opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder', 'opt_state', '1', '1')": {"key_metadata": [{"key": "encoder", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('encoder_target', 'step')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'params', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('encoder_target', 'opt_state')": {"key_metadata": [{"key": "encoder_target", "key_type": 2}, {"key": "opt_state", "key_type": 2}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('latent_model', 'step')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_2', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_1', 'Layer_2', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'params', 'params', 'MLP_2', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '0')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('latent_model', 'opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_2', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_1', 'Layer_2', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'mu', 'params', 'MLP_2', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_2', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_1', 'Layer_2', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_1", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '0', 'nu', 'params', 'MLP_2', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_2", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('latent_model', 'opt_state', '1', '1')": {"key_metadata": [{"key": "latent_model", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('step',)": {"key_metadata": [{"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('entropy', 'step')": {"key_metadata": [{"key": "entropy", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('entropy', 'params', 'log_alpha')": {"key_metadata": [{"key": "entropy", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "log_alpha", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('entropy', 'opt_state', '0')": {"key_metadata": [{"key": "entropy", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('entropy', 'opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "entropy", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('entropy', 'opt_state', '1', '0', 'mu', 'log_alpha')": {"key_metadata": [{"key": "entropy", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "log_alpha", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('entropy', 'opt_state', '1', '0', 'nu', 'log_alpha')": {"key_metadata": [{"key": "entropy", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "log_alpha", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('entropy', 'opt_state', '1', '1')": {"key_metadata": [{"key": "entropy", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('alpha_actor', 'step')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'params', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '0')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('alpha_actor', 'opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'mu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '0', 'nu', 'params', 'MLP_0', 'Layer_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "MLP_0", "key_type": 2}, {"key": "Layer_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('alpha_actor', 'opt_state', '1', '1')": {"key_metadata": [{"key": "alpha_actor", "key_type": 2}, {"key": "opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('reward_norm_std',)": {"key_metadata": [{"key": "reward_norm_std", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('mrq_reward_scale',)": {"key_metadata": [{"key": "mrq_reward_scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('mrq_target_reward_scale',)": {"key_metadata": [{"key": "mrq_target_reward_scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/_sharding b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..bb89520631d50ff3fc78f2295a4b1dfe5db3f6fa
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/_sharding
@@ -0,0 +1 @@
+{"Y3JpdGljLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm11LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm51LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5wYXJhbXMubG9nX2FscGhh":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3Jld2FyZF9zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3RhcmdldF9yZXdhcmRfc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cmV3YXJkX25vcm1fc3Rk":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/d/47d853b1a9a333d0c18a184f391de768 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/d/47d853b1a9a333d0c18a184f391de768
new file mode 100644
index 0000000000000000000000000000000000000000..45e1cd590dd3ec7c5022b993ffdb6bec8ddc736e
Binary files /dev/null and b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/d/47d853b1a9a333d0c18a184f391de768 differ
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/manifest.ocdbt b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2e4cb85f564c079715f01a9e7e911935f32bf42d
Binary files /dev/null and b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/manifest.ocdbt differ
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/1339297efeac8b69a7002d232e87ef72 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/1339297efeac8b69a7002d232e87ef72
new file mode 100644
index 0000000000000000000000000000000000000000..07c1d802758b25e677faa3e4e58d6689df705d5a
Binary files /dev/null and b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/1339297efeac8b69a7002d232e87ef72 differ
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/333ef4658b7da8d91d7eb966cc8f8d61 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/333ef4658b7da8d91d7eb966cc8f8d61
new file mode 100644
index 0000000000000000000000000000000000000000..259844f1b6ce250f2f2d8c800a7d4c943c1b1e60
Binary files /dev/null and b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/333ef4658b7da8d91d7eb966cc8f8d61 differ
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/4210ebb8ca8c9efdfad2f0b95307cfcb b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/4210ebb8ca8c9efdfad2f0b95307cfcb
new file mode 100644
index 0000000000000000000000000000000000000000..c0d342e0ce8e20188ee09880c2eb0f36a218f87f
Binary files /dev/null and b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/4210ebb8ca8c9efdfad2f0b95307cfcb differ
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/4385a586b5625eeb416b5c6a8c4aaad6 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/4385a586b5625eeb416b5c6a8c4aaad6
new file mode 100644
index 0000000000000000000000000000000000000000..17cdecd0f22f83b7175bdecac4b3c3431995f472
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/4385a586b5625eeb416b5c6a8c4aaad6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:68354d5f142dedfa7e606439ec56e199f8995ad8b0ce2d96e536cf75fb7734f9
+size 392338279
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/6d4acf101dafe142d208e5e8f2a8aa58 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/6d4acf101dafe142d208e5e8f2a8aa58
new file mode 100644
index 0000000000000000000000000000000000000000..1d70587dd77191b31e80162ae80b595210ac87d7
Binary files /dev/null and b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/6d4acf101dafe142d208e5e8f2a8aa58 differ
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/85dfee5343855c9cde0310dab8671e60 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/85dfee5343855c9cde0310dab8671e60
new file mode 100644
index 0000000000000000000000000000000000000000..7d938b71926d0d7ff09380e86951bf8bed5181c8
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/85dfee5343855c9cde0310dab8671e60
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c0a84ac22aaa74625f0f8d5c9e2308b4f3058c9eb2354586f788f846d56beff1
+size 113711
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/9bf96da929a70ea41e16bd0d957238c3 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/9bf96da929a70ea41e16bd0d957238c3
new file mode 100644
index 0000000000000000000000000000000000000000..c652139d3562ba54893ae5837d815d31edb05313
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/9bf96da929a70ea41e16bd0d957238c3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30007456070200ddbe19976ef1c238b9a19e69461ee867ff1cc471a3ad49feb9
+size 29530080
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/b25c2843d529ba5ba9bfa8dfd66082ad b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/b25c2843d529ba5ba9bfa8dfd66082ad
new file mode 100644
index 0000000000000000000000000000000000000000..08fbdadc5f228741133005fc96099b6eab06d599
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/b25c2843d529ba5ba9bfa8dfd66082ad
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9fe5ec5c273553725b9219031a4484493d176b965b6417d528d01635353ae293
+size 420285
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/df28d728678e56ae0549ea3bcef7da69 b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/df28d728678e56ae0549ea3bcef7da69
new file mode 100644
index 0000000000000000000000000000000000000000..e2c8176744b03b3ab9b3e92c8d5094272bd8d85e
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/d/df28d728678e56ae0549ea3bcef7da69
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27a7bee091955731aa306fac6813de0c830c7da76938d8e6e735e00789e4ed01
+size 216165288
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/manifest.ocdbt b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..04d4a552341d6413fd7f45d69bd461550a6c5d5f
Binary files /dev/null and b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/1600000/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000013.npz b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000013.npz
new file mode 100644
index 0000000000000000000000000000000000000000..eb974e931e25ea95bc1ba32755c21ea84309a778
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000013.npz
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db7b62cf78678980d3a27b93b724eef4467810f87d925ea1763b1ee76281e1e2
+size 5384660
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000014.npz b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000014.npz
new file mode 100644
index 0000000000000000000000000000000000000000..67f2491f2d50ddba94c75041bbc3aa9eb54a7ad8
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000014.npz
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f671e576127023fea165ca31a4f219aebee5690e0921a4f73faebfeb81fa6491
+size 5413820
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000015.npz b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000015.npz
new file mode 100644
index 0000000000000000000000000000000000000000..f5c2c2c77ddda6b98869ff717d0f55531cc39721
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/eval/buffer/replay_buffer_0000015.npz
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8d97094d2d4df96581c5780bc5014ff28a60ef6a7815177e4a55af7bce8a8e8d
+size 5411660
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/steps_done.txt b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/steps_done.txt
index 765b0d772bce342f9aef53b1105cc7ffa86eaf52..ec1b877bb171971b94204b9ee3785f4597fffd71 100644
--- a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/steps_done.txt
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/steps_done.txt
@@ -1 +1 @@
-1300000
\ No newline at end of file
+1600000
\ No newline at end of file
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000013.npz b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000013.npz
new file mode 100644
index 0000000000000000000000000000000000000000..4eb576d07026730e4c82b0d4fdec100e4aa3de5b
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000013.npz
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7662719949040b0f54577c50e8026ab5f6beb36cc66d49c8c2804efbf1137f7
+size 102619220
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000014.npz b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000014.npz
new file mode 100644
index 0000000000000000000000000000000000000000..03af114c37c3fbbd13b8b90f13edf8356b119e2a
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000014.npz
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f4960ead243c1261244a80559115dc8324f62eaf39871177d83c01465e23edf
+size 102590060
diff --git a/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000015.npz b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000015.npz
new file mode 100644
index 0000000000000000000000000000000000000000..eb4239c4f40665fa531444f70fb8b1f76c8a59b9
--- /dev/null
+++ b/Hopper/AE_MAD_SAC_kappa0p03_maxalpha6/utd2/10seed/job28504867_51/checkpoint/train/buffer/replay_buffer_0000015.npz
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a451a2de096ecb23e10ee15c156dc334f4996576e6b3035a7988ee416d1ec15
+size 102592220