Upload Panda DR model weights with LFS

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+**/d/* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +0,0 @@
----
-license: mit
----

seed_1/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779872073991051832, "commit_timestamp_nsecs": 1779872074614595120, "custom_metadata": {}}

seed_1/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('buffer', 'proprio_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "proprio_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 14]}}, "('buffer', 'privileged_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "privileged_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 2]}}, "('buffer', 'vision_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "vision_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 64, 64, 1]}}, "('buffer', 'action_chunks')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "action_chunks", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 16, 7]}}, "('buffer', 'ptr')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "ptr", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('buffer', 'size')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "size", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('student', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('student', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 2]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 263, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 7]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [270, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 8, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'cls_token')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "cls_token", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'pos_embedding')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 65, 128]}}, "('teacher', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16, 256]}}, "('teacher', 'params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('teacher', 'params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}, "('teacher', 'params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 7]}}, "('teacher', 'params', 'Dense_3', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1]}}, "('teacher', 'params', 'Dense_3', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1]}}, "('teacher', 'params', 'log_std')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "log_std", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

seed_1/default/_sharding

@@ -0,0 +1 @@
+{"YnVmZmVyLmFjdGlvbl9jaHVua3M=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnB0cg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByaXZpbGVnZWRfb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByb3ByaW9fb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnNpemU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnZpc2lvbl9vYnM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLmNsc190b2tlbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLnBvc19lbWJlZGRpbmc=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMubG9nX3N0ZA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

seed_1/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "buffer.proprio_obs", "write_shape": [100000, 14], "chunk_shape": [100000, 14], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.privileged_obs", "write_shape": [100000, 2], "chunk_shape": [100000, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.vision_obs", "write_shape": [100000, 64, 64, 1], "chunk_shape": [100000, 64, 64, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.action_chunks", "write_shape": [100000, 16, 7], "chunk_shape": [100000, 16, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.ptr", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.size", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.kernel", "write_shape": [128, 128], "chunk_shape": [128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_B.kernel", "write_shape": [8, 128], "chunk_shape": [8, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.kernel", "write_shape": [128, 2], "chunk_shape": [128, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_B.kernel", "write_shape": [8, 2], "chunk_shape": [8, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.kernel", "write_shape": [3, 263, 256], "chunk_shape": [3, 263, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.kernel", "write_shape": [3, 256, 256], "chunk_shape": [3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.bias", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.kernel", "write_shape": [3, 256, 7], "chunk_shape": [3, 256, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.kernel", "write_shape": [1, 128], "chunk_shape": [1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.kernel", "write_shape": [270, 128], "chunk_shape": [270, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.kernel", "write_shape": [8, 8, 1, 128], "chunk_shape": [8, 8, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.cls_token", "write_shape": [1, 1, 128], "chunk_shape": [1, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.pos_embedding", "write_shape": [1, 65, 128], "chunk_shape": [1, 65, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.kernel", "write_shape": [16, 256], "chunk_shape": [16, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.kernel", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.bias", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.kernel", "write_shape": [256, 7], "chunk_shape": [256, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.kernel", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.log_std", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}]}

seed_1/default/d/6946cac8a54467ac7fabfd6fd3c0ece6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b4b903c44697127edafb01ce55ece1e5d88e0a71da440378775687148afc2f0
+size 23424

seed_1/default/ocdbt.process_0/d/290957ed8305a3a33d825bf0b19ed857

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f811ec5d6385a243d94c05748e6084ea2586f984db80af4b9548493567ca7710
+size 1059

seed_1/default/ocdbt.process_0/d/34c33b3868b53bfaeb394cb6a7aa9eec

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d7975b812c1abdffde014c400566ec4455ae4a8acfe1f3777eb28694aca7a7a
+size 19030016

seed_1/default/ocdbt.process_0/d/77cba3ff7adaa70c14c8f86b6bcc2dad

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2bd60c14a16965d78353bb0f4df2b2ad8461d496598f60b9197a83086b3ed61c
+size 73425

seed_1/default/ocdbt.process_0/d/c2ecfe12e0da1eaf6b10f0c36af59bd9

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2167e794a2e6f0149b9e4997153d382dda5cad1c216219b70c7d6f347eb77d77
+size 1995352

seed_1/default/ocdbt.process_0/d/d975a55baf59c4e90fd548ce0ad2c411

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8bf56dd5dffe65ddc9366735a4be803374dddb66542b356a77444f85400d920e
+size 263

seed_2/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779874426302981773, "commit_timestamp_nsecs": 1779874426922654015, "custom_metadata": {}}

seed_2/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('buffer', 'proprio_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "proprio_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 14]}}, "('buffer', 'privileged_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "privileged_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 2]}}, "('buffer', 'vision_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "vision_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 64, 64, 1]}}, "('buffer', 'action_chunks')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "action_chunks", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 16, 7]}}, "('buffer', 'ptr')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "ptr", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('buffer', 'size')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "size", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('student', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('student', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 2]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 263, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 7]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [270, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 8, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'cls_token')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "cls_token", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'pos_embedding')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 65, 128]}}, "('teacher', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16, 256]}}, "('teacher', 'params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('teacher', 'params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}, "('teacher', 'params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 7]}}, "('teacher', 'params', 'Dense_3', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1]}}, "('teacher', 'params', 'Dense_3', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1]}}, "('teacher', 'params', 'log_std')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "log_std", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

seed_2/default/_sharding

@@ -0,0 +1 @@
+{"YnVmZmVyLmFjdGlvbl9jaHVua3M=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnB0cg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByaXZpbGVnZWRfb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByb3ByaW9fb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnNpemU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnZpc2lvbl9vYnM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLmNsc190b2tlbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLnBvc19lbWJlZGRpbmc=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMubG9nX3N0ZA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

seed_2/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "buffer.proprio_obs", "write_shape": [100000, 14], "chunk_shape": [100000, 14], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.privileged_obs", "write_shape": [100000, 2], "chunk_shape": [100000, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.vision_obs", "write_shape": [100000, 64, 64, 1], "chunk_shape": [100000, 64, 64, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.action_chunks", "write_shape": [100000, 16, 7], "chunk_shape": [100000, 16, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.ptr", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.size", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.kernel", "write_shape": [128, 128], "chunk_shape": [128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_B.kernel", "write_shape": [8, 128], "chunk_shape": [8, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.kernel", "write_shape": [128, 2], "chunk_shape": [128, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_B.kernel", "write_shape": [8, 2], "chunk_shape": [8, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.kernel", "write_shape": [3, 263, 256], "chunk_shape": [3, 263, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.kernel", "write_shape": [3, 256, 256], "chunk_shape": [3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.bias", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.kernel", "write_shape": [3, 256, 7], "chunk_shape": [3, 256, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.kernel", "write_shape": [1, 128], "chunk_shape": [1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.kernel", "write_shape": [270, 128], "chunk_shape": [270, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.kernel", "write_shape": [8, 8, 1, 128], "chunk_shape": [8, 8, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.cls_token", "write_shape": [1, 1, 128], "chunk_shape": [1, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.pos_embedding", "write_shape": [1, 65, 128], "chunk_shape": [1, 65, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.kernel", "write_shape": [16, 256], "chunk_shape": [16, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.kernel", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.bias", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.kernel", "write_shape": [256, 7], "chunk_shape": [256, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.kernel", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.log_std", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}]}

seed_2/default/d/a26856a36853bf9ae80be12de123c72f

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:93c251d1d97e7aba18ef595f82861a4540f36105f47865305ce50605194a8329
+size 23402

seed_2/default/ocdbt.process_0/d/1fa8878c8ab0411a63c07079901af9ce

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76f4498c77ae69db10afc5684239c42cb11f90cb463753c3acdf73badda95be7
+size 73403

seed_2/default/ocdbt.process_0/d/2e3e772bdeb9ae842df6881626d05ab7

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bac75654cd50ca2185781698e235a0486c357fa0f1f0d6206e43c229245b2194
+size 296

seed_2/default/ocdbt.process_0/d/5d6db9124086e6c7fa0d7ca3936dd35d

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f811ec5d6385a243d94c05748e6084ea2586f984db80af4b9548493567ca7710
+size 1059

seed_2/default/ocdbt.process_0/d/84a9dda5641f2d611525f08ae2d8cc8c

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2d07cde92685917ffbe60125e267796550882cacfb1a70c6d7a7b2f78b899cbc
+size 20008960

seed_2/default/ocdbt.process_0/d/a6d8978bd3db532722a9cb32a692b115

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:73424aa5c9555bc2aff3e0d0d8dfcf835f2b29180a922d23225532b35c10be82
+size 266515

seed_42/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779875372514651978, "commit_timestamp_nsecs": 1779875373242780425, "custom_metadata": {}}

seed_42/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('buffer', 'proprio_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "proprio_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 14]}}, "('buffer', 'privileged_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "privileged_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 2]}}, "('buffer', 'vision_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "vision_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 64, 64, 1]}}, "('buffer', 'action_chunks')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "action_chunks", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 16, 7]}}, "('buffer', 'ptr')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "ptr", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('buffer', 'size')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "size", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('student', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('student', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 2]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 2]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 263, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 7]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [270, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 8, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'cls_token')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "cls_token", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'pos_embedding')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 65, 128]}}, "('teacher', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16, 256]}}, "('teacher', 'params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('teacher', 'params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}, "('teacher', 'params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 7]}}, "('teacher', 'params', 'Dense_3', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1]}}, "('teacher', 'params', 'Dense_3', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1]}}, "('teacher', 'params', 'log_std')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "log_std", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7]}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

seed_42/default/_sharding

@@ -0,0 +1 @@
+{"YnVmZmVyLmFjdGlvbl9jaHVua3M=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnB0cg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByaXZpbGVnZWRfb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByb3ByaW9fb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnNpemU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnZpc2lvbl9vYnM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLmNsc190b2tlbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLnBvc19lbWJlZGRpbmc=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMubG9nX3N0ZA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

seed_42/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "buffer.proprio_obs", "write_shape": [100000, 14], "chunk_shape": [100000, 14], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.privileged_obs", "write_shape": [100000, 2], "chunk_shape": [100000, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.vision_obs", "write_shape": [100000, 64, 64, 1], "chunk_shape": [100000, 64, 64, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.action_chunks", "write_shape": [100000, 16, 7], "chunk_shape": [100000, 16, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.ptr", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.size", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.kernel", "write_shape": [128, 128], "chunk_shape": [128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_B.kernel", "write_shape": [8, 128], "chunk_shape": [8, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.kernel", "write_shape": [128, 2], "chunk_shape": [128, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_B.kernel", "write_shape": [8, 2], "chunk_shape": [8, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.kernel", "write_shape": [3, 263, 256], "chunk_shape": [3, 263, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.kernel", "write_shape": [3, 256, 256], "chunk_shape": [3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.bias", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.kernel", "write_shape": [3, 256, 7], "chunk_shape": [3, 256, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.kernel", "write_shape": [1, 128], "chunk_shape": [1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.kernel", "write_shape": [270, 128], "chunk_shape": [270, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.kernel", "write_shape": [8, 8, 1, 128], "chunk_shape": [8, 8, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.cls_token", "write_shape": [1, 1, 128], "chunk_shape": [1, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.pos_embedding", "write_shape": [1, 65, 128], "chunk_shape": [1, 65, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.kernel", "write_shape": [16, 256], "chunk_shape": [16, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.kernel", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.bias", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.kernel", "write_shape": [256, 7], "chunk_shape": [256, 7], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.kernel", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.log_std", "write_shape": [7], "chunk_shape": [7], "ext_metadata": null}}]}

seed_42/default/d/26ab95f9c110484cb46e3a400b960d1e

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee094aee0b39a7578ebfd7b4271d9c69b82072ba3619e8ab832ec4ed6435b329
+size 23478

seed_42/default/ocdbt.process_0/d/12e15869bd66930234e0aa7ea1c81ee5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:145f685a3c40b2605cf2143d544565e45a35438d60177be9e5e5291253e3dfa2
+size 15843328

seed_42/default/ocdbt.process_0/d/15b63973ddab74a85b5a28317be6798d

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f811ec5d6385a243d94c05748e6084ea2586f984db80af4b9548493567ca7710
+size 1059

seed_42/default/ocdbt.process_0/d/1bf8564de080bc2477cc10bf81ec1ef6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d414cbd0a42997c28c03e725445025b75a00b8c5c1180ff2f54df53e40376fa9
+size 3276331

seed_42/default/ocdbt.process_0/d/2e5e89e24aa6cd2fbd81cef734addd10

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2aed0ec78592df75aae262ff704e2ede9b5fde74e1adad2649786a422c733143
+size 1604087

seed_42/default/ocdbt.process_0/d/60d74638ae1e997ffdd96e99bb9ec530

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6f4f6020076bc217cd229c0a02898ae494080f0c7b3bae78baa473d6d181d29
+size 318

seed_42/default/ocdbt.process_0/d/6a6357cb427e674ab8d5724d5c249f3e

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:322e08f5189be7a579446bd8c28015144c8cda7a6b1714cbaa5dbe5a4add2544
+size 73480