Add files using upload-large-folder tool

.gitattributes

@@ -54,3 +54,23 @@ robocasa_test_all_atomic/2000/params/ocdbt.process_0/d/3a09fc65139068f2492acf80d
 robocasa_test_all_atomic/2000/params/ocdbt.process_0/d/24a93e48bc2a9d750cc558de4f4c8449 filter=lfs diff=lfs merge=lfs -text
 robocasa_test_all_atomic/2000/params/ocdbt.process_0/d/afcfa0b4f1390292c94e5be65d4e1dc4 filter=lfs diff=lfs merge=lfs -text
 robocasa_test_all_atomic/2000/params/ocdbt.process_0/d/08fb9a1f12f1ba08b009af42746e9d6c filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/fed26169091de6ebe665b5f1650a32e4 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/8a1a31080a003651325b53101602a222 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/11d91195b7ccdcb9fcc38286025798f0 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/5ad9fe5d7623dd27183833b3506e97e0 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/c23818a6c00b9dce79f344566f1664db filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/6c4a6c89c9d9398d61cd0da6cf1a90ac filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/67bf790fde79982c4b2440578901b080 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/c91978c4f15685f7d2b9923b4b71cccf filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/7488a0e72d0e1c02b16f65eef5311b33 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/4d9fbea72ecd6c791646988839d5a1d6 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/d14284e5b6addb7b1e4b3a523b1b3abd filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/b48900465bbb40513b85d3fd3f7585bc filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/51f9a35e510485b097a245da6d1850b0 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/ad5b192d98866d982f62ed8fb6a35575 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/8a67b6e3e0e38da7eaf6e7931ccdbac1 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/2b10a95bb24cbcfbb7978900c6770289 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/874c622005ab2d93d55276b01e6d693a filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/a19ff54303520b45679d3b8c6816f0b3 filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/80f139e7ec0837ca6e1ab879c237ebfa filter=lfs diff=lfs merge=lfs -text
+robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/0b91f5a8830aee96356a8c25f29c9f4c filter=lfs diff=lfs merge=lfs -text

robocasa_test_all_atomic/5000/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"assets": "openpi.training.checkpoints.CallbackHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1775490666314870914, "commit_timestamp_nsecs": 1775490712190548573, "custom_metadata": {}}

robocasa_test_all_atomic/5000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json

@@ -0,0 +1,464 @@
+{
+  "norm_stats": {
+    "state": {
+      "mean": [
+        2.637775182723999,
+        -2.0912017822265625,
+        0.7007339596748352,
+        0.0,
+        0.0,
+        0.24000521004199982,
+        0.6076071262359619,
+        0.258401483297348,
+        -0.018448200076818466,
+        0.44676560163497925,
+        -0.26384344696998596,
+        0.03587118536233902,
+        -0.07241316884756088,
+        0.13728724420070648,
+        0.03210829943418503,
+        -0.03232070058584213
+      ],
+      "std": [
+        1.6751232147216797,
+        1.3338454961776733,
+        0.0,
+        0.0,
+        0.0,
+        0.6626312732696533,
+        0.3660806715488434,
+        0.16573558747768402,
+        0.2436121702194214,
+        0.2089885175228119,
+        0.8243448734283447,
+        0.29920220375061035,
+        0.3282162845134735,
+        0.1679685264825821,
+        0.012291335500776768,
+        0.012060738168656826
+      ],
+      "q01": [
+        -1.218368652534485,
+        -5.65514452624917,
+        0.6999506360054016,
+        0.0,
+        0.0,
+        -1.0,
+        0.0,
+        -0.13958552385568623,
+        -0.5809329965710639,
+        -0.1109548017382622,
+        -0.9983834087371826,
+        -0.6892688079476357,
+        -0.745897297000885,
+        0.0007815830585608637,
+        -0.000049362923577427054,
+        -0.04081762075200677
+      ],
+      "q99": [
+        6.3111974027633675,
+        -0.46855325088500965,
+        0.7051179245471955,
+        0.0,
+        0.0,
+        0.9996,
+        0.9998,
+        0.7251633455753326,
+        0.5839828933954241,
+        0.8639214414954184,
+        0.9963686748266221,
+        0.7791513680815696,
+        0.726318956565857,
+        0.6806885522310082,
+        0.040635910079255703,
+        0.00001817415971308578
+      ]
+    },
+    "actions": {
+      "mean": [
+        0.026344629004597664,
+        0.0013832177501171827,
+        -0.0021026874892413616,
+        0.0,
+        -0.8568487763404846,
+        -0.009222988970577717,
+        -0.0020607062615454197,
+        -0.051031820476055145,
+        0.008847072720527649,
+        -0.018305813893675804,
+        -0.0024289467837661505,
+        -0.330047070980072
+      ],
+      "std": [
+        0.16712690889835358,
+        0.1392730474472046,
+        0.12041067332029343,
+        0.0,
+        0.5155678391456604,
+        0.4523046910762787,
+        0.38781222701072693,
+        0.4078434407711029,
+        0.11646749824285507,
+        0.13211272656917572,
+        0.1214357316493988,
+        0.9439644813537598
+      ],
+      "q01": [
+        -0.0716,
+        -0.5571999999999999,
+        -0.5371999999999999,
+        0.0,
+        -1.0,
+        -1.0,
+        -1.0,
+        -1.0,
+        -0.346,
+        -0.45719999999999994,
+        -0.4032,
+        -1.0
+      ],
+      "q99": [
+        0.9996,
+        0.6356000000000002,
+        0.4828000000000001,
+        0.0,
+        0.9996,
+        0.9996,
+        0.9996,
+        0.9996,
+        0.4456,
+        0.3740000000000001,
+        0.3884000000000001,
+        0.9996
+      ]
+    }
+  },
+  "sample_ratio": 0.05,
+  "dataset_infos": {
+    "robotgeneralist/AdjustToasterOvenTemperature_pretrain": {
+      "commit_hash": "24364b1b9664fe7a58ecfce7e61cac66bcde1209",
+      "num_episodes": 107,
+      "num_frames": 21328
+    },
+    "robotgeneralist/AdjustWaterTemperature_pretrain": {
+      "commit_hash": "f1bcd907c0510f687521a44b98edaba314548a95",
+      "num_episodes": 106,
+      "num_frames": 20953
+    },
+    "robotgeneralist/CheesyBread_pretrain": {
+      "commit_hash": "d59106dabf8827be89ea3513399b6a48c8cbf814",
+      "num_episodes": 101,
+      "num_frames": 31141
+    },
+    "robotgeneralist/CloseBlenderLid_pretrain": {
+      "commit_hash": "68fb8f0d7495e9da1382c803467c488b97b55bf0",
+      "num_episodes": 106,
+      "num_frames": 36933
+    },
+    "robotgeneralist/CloseCabinet_pretrain": {
+      "commit_hash": "b58bca4c52b1673fb6a20fd0a15e9031f5b3eb72",
+      "num_episodes": 105,
+      "num_frames": 27754
+    },
+    "robotgeneralist/CloseDishwasher_pretrain": {
+      "commit_hash": "659c06a9486fb1a0eaf37c25b35af0b3e8ff2d4e",
+      "num_episodes": 111,
+      "num_frames": 15781
+    },
+    "robotgeneralist/CloseDrawer_pretrain": {
+      "commit_hash": "ab571fcde78de907f8154bcad20101423a8ee503",
+      "num_episodes": 110,
+      "num_frames": 15670
+    },
+    "robotgeneralist/CloseElectricKettleLid_pretrain": {
+      "commit_hash": "b4b4c5967fe877790ac86c5bf9e8a0216e229105",
+      "num_episodes": 102,
+      "num_frames": 7530
+    },
+    "robotgeneralist/CloseFridge_pretrain": {
+      "commit_hash": "fb04ab4ebc39a0a23dc27140f4f7701b91b7313b",
+      "num_episodes": 106,
+      "num_frames": 26888
+    },
+    "robotgeneralist/CloseFridgeDrawer_pretrain": {
+      "commit_hash": "ed9f1b467346f600ae6f755ebc7ad0e1dd841dda",
+      "num_episodes": 111,
+      "num_frames": 14946
+    },
+    "robotgeneralist/CloseMicrowave_pretrain": {
+      "commit_hash": "61e4ea6d29e504329df7147ca563d34324ae7aa5",
+      "num_episodes": 105,
+      "num_frames": 20075
+    },
+    "robotgeneralist/CloseOven_pretrain": {
+      "commit_hash": "61b80872b74919655d4fed2fc4e8667ce248fa48",
+      "num_episodes": 107,
+      "num_frames": 18230
+    },
+    "robotgeneralist/CloseStandMixerHead_pretrain": {
+      "commit_hash": "855c0a58a8396f8855eb8e213e6bed53dcde551a",
+      "num_episodes": 110,
+      "num_frames": 11593
+    },
+    "robotgeneralist/CloseToasterOvenDoor_pretrain": {
+      "commit_hash": "3c16da6f363ad8fd2bcd811efd4a87c6ed61a82c",
+      "num_episodes": 110,
+      "num_frames": 19815
+    },
+    "robotgeneralist/CoffeeServeMug_pretrain": {
+      "commit_hash": "23157f307108f6e352b98b536e941d1086c3d8af",
+      "num_episodes": 108,
+      "num_frames": 16921
+    },
+    "robotgeneralist/CoffeeSetupMug_pretrain": {
+      "commit_hash": "76e51e454262ddeb046b015486c1fe4602a53b49",
+      "num_episodes": 105,
+      "num_frames": 23636
+    },
+    "robotgeneralist/LowerHeat_pretrain": {
+      "commit_hash": "891f233633a61ce11f4fefe95a3b5081657c0227",
+      "num_episodes": 110,
+      "num_frames": 31174
+    },
+    "robotgeneralist/MakeIcedCoffee_pretrain": {
+      "commit_hash": "aee5552cade8433a5c9282851250ee0a7c6c4717",
+      "num_episodes": 102,
+      "num_frames": 29048
+    },
+    "robotgeneralist/NavigateKitchen_pretrain": {
+      "commit_hash": "9df39eed58424fbe7945717c54549e5308c9fcc6",
+      "num_episodes": 503,
+      "num_frames": 79550
+    },
+    "robotgeneralist/OpenBlenderLid_pretrain": {
+      "commit_hash": "302d625654bdb678e877dff9b6a0d9c9dcc3fc15",
+      "num_episodes": 106,
+      "num_frames": 20124
+    },
+    "robotgeneralist/OpenCabinet_pretrain": {
+      "commit_hash": "ee50aad81bc9a115a5f00885d45046d6539646cf",
+      "num_episodes": 107,
+      "num_frames": 37492
+    },
+    "robotgeneralist/OpenDishwasher_pretrain": {
+      "commit_hash": "7dd0903b5ddf482ddedbd0f013fe5223118d47b2",
+      "num_episodes": 111,
+      "num_frames": 18086
+    },
+    "robotgeneralist/OpenDrawer_pretrain": {
+      "commit_hash": "f2abb58f7fdf0157aeeaa645791a460de44be1d8",
+      "num_episodes": 102,
+      "num_frames": 20488
+    },
+    "robotgeneralist/OpenElectricKettleLid_pretrain": {
+      "commit_hash": "5416f530c440f61466d02f6ce0bf549973c0db95",
+      "num_episodes": 110,
+      "num_frames": 10928
+    },
+    "robotgeneralist/OpenFridge_pretrain": {
+      "commit_hash": "5869139ca48dbe66f7d8dabe51d4ecaf4fb71c9d",
+      "num_episodes": 105,
+      "num_frames": 33138
+    },
+    "robotgeneralist/OpenFridgeDrawer_pretrain": {
+      "commit_hash": "ceda3eb991f74682c8bc679c348402ea4587f5c4",
+      "num_episodes": 107,
+      "num_frames": 18517
+    },
+    "robotgeneralist/OpenMicrowave_pretrain": {
+      "commit_hash": "6222929e60bbee3d135cfdc27d6663e3fc535ead",
+      "num_episodes": 105,
+      "num_frames": 26017
+    },
+    "robotgeneralist/OpenOven_pretrain": {
+      "commit_hash": "b8e0e83300e3754fdd21de46680cb2852d8905c9",
+      "num_episodes": 108,
+      "num_frames": 15555
+    },
+    "robotgeneralist/OpenStandMixerHead_pretrain": {
+      "commit_hash": "43c2fe25f56fc1904f12db1b66855c1dd422463a",
+      "num_episodes": 109,
+      "num_frames": 13411
+    },
+    "robotgeneralist/OpenToasterOvenDoor_pretrain": {
+      "commit_hash": "ad2164f3590a05d06ba5cd2aea8216ba73619cb6",
+      "num_episodes": 105,
+      "num_frames": 15469
+    },
+    "robotgeneralist/PackDessert_pretrain": {
+      "commit_hash": "63d48fe23f8fb5d5a9f603b62c38f22ad0d5a418",
+      "num_episodes": 108,
+      "num_frames": 27994
+    },
+    "robotgeneralist/PickPlaceCabinetToCounter_pretrain": {
+      "commit_hash": "a338e9f408bad681248461fe42deffb77e0ab67a",
+      "num_episodes": 106,
+      "num_frames": 20201
+    },
+    "robotgeneralist/PickPlaceCounterToBlender_pretrain": {
+      "commit_hash": "6a2aba5d6a1fe63f65f7ad40d5098e8d68b08e72",
+      "num_episodes": 106,
+      "num_frames": 38892
+    },
+    "robotgeneralist/PickPlaceCounterToCabinet_pretrain": {
+      "commit_hash": "628a13d9c15c967e6133d394913b0dfe55e8fa8e",
+      "num_episodes": 108,
+      "num_frames": 24225
+    },
+    "robotgeneralist/PickPlaceCounterToDrawer_pretrain": {
+      "commit_hash": "35ac4c271761916cde68099be0fad4f59ad68dba",
+      "num_episodes": 107,
+      "num_frames": 28225
+    },
+    "robotgeneralist/PickPlaceCounterToMicrowave_pretrain": {
+      "commit_hash": "c9720edf6282bcf99f2a76505804f81d4ac97e92",
+      "num_episodes": 110,
+      "num_frames": 42012
+    },
+    "robotgeneralist/PickPlaceCounterToOven_pretrain": {
+      "commit_hash": "f1d9b1c899de4abec2f3a11b5abd0db519f555d7",
+      "num_episodes": 111,
+      "num_frames": 32014
+    },
+    "robotgeneralist/PickPlaceCounterToSink_pretrain": {
+      "commit_hash": "8642bb8c7db762fe4b82fd0b0e26847223f83de0",
+      "num_episodes": 108,
+      "num_frames": 22410
+    },
+    "robotgeneralist/PickPlaceCounterToStandMixer_pretrain": {
+      "commit_hash": "76ca30ee317ae85519afb52a0bec1ee8d116a732",
+      "num_episodes": 107,
+      "num_frames": 25467
+    },
+    "robotgeneralist/PickPlaceCounterToStove_pretrain": {
+      "commit_hash": "bf5a50869a7ecc1050f878001b39c6cb3bb0bfe5",
+      "num_episodes": 108,
+      "num_frames": 24039
+    },
+    "robotgeneralist/PickPlaceCounterToToasterOven_pretrain": {
+      "commit_hash": "515f871b5d014f070bb221d7cd9568508db890ef",
+      "num_episodes": 108,
+      "num_frames": 24313
+    },
+    "robotgeneralist/PickPlaceDrawerToCounter_pretrain": {
+      "commit_hash": "cf62a0e481b087d531fc8cc6b49f1d697ac82d9e",
+      "num_episodes": 103,
+      "num_frames": 31819
+    },
+    "robotgeneralist/PickPlaceFridgeDrawerToShelf_pretrain": {
+      "commit_hash": "b944b9dd65fb12c6253e3254f0acd7bfb7af730f",
+      "num_episodes": 106,
+      "num_frames": 26396
+    },
+    "robotgeneralist/PickPlaceFridgeShelfToDrawer_pretrain": {
+      "commit_hash": "5381c4c9ff0e1a09cf963d537836e1060ef75565",
+      "num_episodes": 111,
+      "num_frames": 27047
+    },
+    "robotgeneralist/PickPlaceMicrowaveToCounter_pretrain": {
+      "commit_hash": "de6cc72aa95974b46fd965d66ab60629c68503a3",
+      "num_episodes": 112,
+      "num_frames": 38729
+    },
+    "robotgeneralist/PickPlaceSinkToCounter_pretrain": {
+      "commit_hash": "854d27642b1810e7576cefe465c181bbee5dfcb3",
+      "num_episodes": 108,
+      "num_frames": 26397
+    },
+    "robotgeneralist/PickPlaceStoveToCounter_pretrain": {
+      "commit_hash": "fca1c62a4f719a24a0e8b587c0422e37c7ef5756",
+      "num_episodes": 109,
+      "num_frames": 23003
+    },
+    "robotgeneralist/PickPlaceToasterOvenToCounter_pretrain": {
+      "commit_hash": "18f6af5e40443639cb1af2e32cb991d8813e016e",
+      "num_episodes": 106,
+      "num_frames": 19323
+    },
+    "robotgeneralist/PickPlaceToasterToCounter_pretrain": {
+      "commit_hash": "f748ad7b5f13a1faec1fb21a8640903a344298ba",
+      "num_episodes": 105,
+      "num_frames": 26907
+    },
+    "robotgeneralist/PreheatOven_pretrain": {
+      "commit_hash": "fe6d8d7f32aabfb9605daec9d6dc9e5552222f43",
+      "num_episodes": 107,
+      "num_frames": 21102
+    },
+    "robotgeneralist/SlideDishwasherRack_pretrain": {
+      "commit_hash": "ee2be96cab8d45b0713a4d3f4c77b7d086b593e7",
+      "num_episodes": 100,
+      "num_frames": 19052
+    },
+    "robotgeneralist/SlideOvenRack_pretrain": {
+      "commit_hash": "ec897192c4630505dbf7ecafc87672d945b5806b",
+      "num_episodes": 113,
+      "num_frames": 23958
+    },
+    "robotgeneralist/SlideToasterOvenRack_pretrain": {
+      "commit_hash": "90c2db6d3c847d2a25ced10ba441cd79845218d3",
+      "num_episodes": 105,
+      "num_frames": 11496
+    },
+    "robotgeneralist/StartCoffeeMachine_pretrain": {
+      "commit_hash": "62b11dc4fa1459dffac7c20f596a282181f8ebba",
+      "num_episodes": 108,
+      "num_frames": 13722
+    },
+    "robotgeneralist/TurnOffMicrowave_pretrain": {
+      "commit_hash": "b40bb9ef7cdb8dd9c5da4487b3d4e538016d28ee",
+      "num_episodes": 108,
+      "num_frames": 15233
+    },
+    "robotgeneralist/TurnOffSinkFaucet_pretrain": {
+      "commit_hash": "4726894e94f3b4baf80a96cc9041eb95fe399fb1",
+      "num_episodes": 106,
+      "num_frames": 12309
+    },
+    "robotgeneralist/TurnOffStove_pretrain": {
+      "commit_hash": "c5029366ce8dd5becc73980076d4e6106d76dc76",
+      "num_episodes": 109,
+      "num_frames": 32741
+    },
+    "robotgeneralist/TurnOnBlender_pretrain": {
+      "commit_hash": "aaa1396e005c3d83f3fa320f3b66a8c240a1e86d",
+      "num_episodes": 107,
+      "num_frames": 11698
+    },
+    "robotgeneralist/TurnOnElectricKettle_pretrain": {
+      "commit_hash": "01adea2d774b56c2f269380b938dc822bd446552",
+      "num_episodes": 108,
+      "num_frames": 12460
+    },
+    "robotgeneralist/TurnOnMicrowave_pretrain": {
+      "commit_hash": "0fd65a1926ad490725cee73933917d7a9985cf57",
+      "num_episodes": 107,
+      "num_frames": 14010
+    },
+    "robotgeneralist/TurnOnSinkFaucet_pretrain": {
+      "commit_hash": "d4efe49ea58fde3cef15480faf663b17af6481f4",
+      "num_episodes": 107,
+      "num_frames": 23795
+    },
+    "robotgeneralist/TurnOnStove_pretrain": {
+      "commit_hash": "782bec492f509094e3dd61938b4d906db62a1321",
+      "num_episodes": 104,
+      "num_frames": 17910
+    },
+    "robotgeneralist/TurnOnToaster_pretrain": {
+      "commit_hash": "5b00e0998c7270bc7b1e106b6c0acd78ee15eac3",
+      "num_episodes": 108,
+      "num_frames": 10042
+    },
+    "robotgeneralist/TurnOnToasterOven_pretrain": {
+      "commit_hash": "f6808d51f8efc6191cf5d1d2d8e795d68657567a",
+      "num_episodes": 103,
+      "num_frames": 17051
+    },
+    "robotgeneralist/TurnSinkSpout_pretrain": {
+      "commit_hash": "504c7c807a1cf800867437f50b6fb08410121807",
+      "num_episodes": 109,
+      "num_frames": 11130
+    }
+  }
+}

robocasa_test_all_atomic/5000/params/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 4304]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 4304]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 4304, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'embedding', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'embedding', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [14, 14, 3, 1152]}}, "('params', 'PaliGemma', 'img', 'head', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048]}}, "('params', 'PaliGemma', 'img', 'head', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152, 2048]}}, "('params', 'PaliGemma', 'img', 'pos_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 256, 1152]}}, "('params', 'PaliGemma', 'llm', 'embedder', 'input_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "embedder", "key_type": 2}, {"key": "input_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [257152, 2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 256, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 256, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 2048, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 1024, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 2048, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 1024, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 2048, 16384]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 16384, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1024, 4096]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 4096, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 1024]}}, "('params', 'action_in_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32, 1024]}}, "('params', 'action_out_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', 'action_out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024, 32]}}, "('params', 'action_time_mlp_in', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_time_mlp_in', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048, 1024]}}, "('params', 'action_time_mlp_out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_time_mlp_out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024, 1024]}}, "('params', 'state_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'state_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

robocasa_test_all_atomic/5000/params/_sharding

@@ -0,0 +1 @@
+{"cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcucG9zX2VtYmVkZGluZy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZW1iZWRkZXIuaW5wdXRfZW1iZWRkaW5nLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5nYXRpbmdfZWluc3VtLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5saW5lYXIudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmdhdGluZ19laW5zdW0udmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmxpbmVhci52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW0udy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW1fMS53LnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2ouYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}"}

robocasa_test_all_atomic/5000/params/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]}

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/0b91f5a8830aee96356a8c25f29c9f4c

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/11d91195b7ccdcb9fcc38286025798f0

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/2b10a95bb24cbcfbb7978900c6770289

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/4d9fbea72ecd6c791646988839d5a1d6

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/51f9a35e510485b097a245da6d1850b0

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/5ad9fe5d7623dd27183833b3506e97e0

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/67bf790fde79982c4b2440578901b080

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/6c4a6c89c9d9398d61cd0da6cf1a90ac

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/7488a0e72d0e1c02b16f65eef5311b33

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/80f139e7ec0837ca6e1ab879c237ebfa

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/874c622005ab2d93d55276b01e6d693a

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/8a1a31080a003651325b53101602a222

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/8a67b6e3e0e38da7eaf6e7931ccdbac1

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/a19ff54303520b45679d3b8c6816f0b3

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/ad5b192d98866d982f62ed8fb6a35575

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/b48900465bbb40513b85d3fd3f7585bc

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/c23818a6c00b9dce79f344566f1664db

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/c91978c4f15685f7d2b9923b4b71cccf

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/d14284e5b6addb7b1e4b3a523b1b3abd

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

robocasa_test_all_atomic/5000/params/ocdbt.process_0/d/fed26169091de6ebe665b5f1650a32e4

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