Add files using upload-large-folder tool

pp__ee_fourier_k50/10000/_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": 1767263708960256543, "commit_timestamp_nsecs": 1767263756536770897, "custom_metadata": {}}

pp__ee_fourier_k50/10000/assets/ur10e/norm_stats.json

@@ -0,0 +1,1525 @@
+{
+  "norm_stats": {
+    "state": {
+      "mean": [
+        0.05313729867339134,
+        0.0037223484832793474,
+        0.106409952044487,
+        0.0008666226058267057,
+        -0.00854077935218811,
+        0.01460699550807476,
+        0.5940247178077698
+      ],
+      "std": [
+        0.1176244392991066,
+        0.16962096095085144,
+        0.07773701101541519,
+        0.04537155479192734,
+        0.06616315245628357,
+        0.6038344502449036,
+        0.49107977747917175
+      ],
+      "q01": [
+        -0.2025573879778385,
+        -0.31106759839057924,
+        0.00890737680345774,
+        -0.11727181684970855,
+        -0.16711108471155167,
+        -1.423702568435669,
+        0.0
+      ],
+      "q99": [
+        0.29901714037060734,
+        0.4115018085956573,
+        0.3217185490995646,
+        0.13774808425903318,
+        0.19241279754638663,
+        1.598234023094177,
+        0.9998
+      ]
+    },
+    "actions": {
+      "mean": [
+        0.0004927651607431471,
+        0.0041174935176968575,
+        0.005378079600632191,
+        -0.0004851108242291957,
+        -0.0010051705176010728,
+        -0.006783927325159311,
+        0.5936462879180908,
+        0.0005593731766566634,
+        -0.00140193453989923,
+        -0.002942675491794944,
+        0.00041861183126457036,
+        0.0004222682327963412,
+        0.0030385160353034735,
+        -0.000329385424265638,
+        -0.0005693831481039524,
+        -0.00029180958517827094,
+        0.000576534541323781,
+        -0.00015302293468266726,
+        -2.4814380594762042e-6,
+        -0.00026009284192696214,
+        0.0003946665092371404,
+        0.000059152920584892854,
+        -0.00006962364568607882,
+        -0.0003498161968309432,
+        0.000053506064432440326,
+        0.00006277022475842386,
+        0.0003592480788938701,
+        -0.00016934213635977358,
+        -0.0001320719748036936,
+        -0.00006360372208291665,
+        0.0001631096820347011,
+        -0.00003750254472834058,
+        -0.000011555765922821593,
+        -0.00006809693149989471,
+        0.00015906375483609736,
+        0.000027621259505394846,
+        -0.00002624911030579824,
+        -0.00012353967758826911,
+        0.000012651072211156134,
+        0.000011669268133118749,
+        0.00012300137314014137,
+        0.00015041981532704085,
+        -0.000056855304137570783,
+        -0.000029746257496299222,
+        0.00007242687570396811,
+        -0.000018986645955010317,
+        -4.556677140499232e-6,
+        -0.00003111416663159616,
+        -0.00009099271846935153,
+        0.000012571626029966865,
+        -0.000013306255823408719,
+        -0.00006410608330043033,
+        6.341869266179856e-6,
+        7.46843352317228e-6,
+        0.00006183652294566855,
+        5.868077096238267e-6,
+        -0.000031927749660098925,
+        -0.00001673644146649167,
+        0.00004087621346116066,
+        -0.000010321637091692537,
+        -2.42824648921669e-6,
+        -0.00001607570084161125,
+        0.00004234428342897445,
+        7.407979410345433e-6,
+        -8.161020559782628e-6,
+        -0.00003870665022986941,
+        3.6879464460071176e-6,
+        4.935619017487625e-6,
+        0.00003818826371571049,
+        0.000051191724196542054,
+        -0.000020595705791492946,
+        -0.000010875675798160955,
+        0.000025950239432859235,
+        -6.903889698151033e-6,
+        -1.2846402341892826e-6,
+        -9.873656381387264e-6,
+        0.00005320658965501934,
+        4.920636911265319e-6,
+        -5.623744073091075e-6,
+        -0.000025824952899711207,
+        2.285133632540237e-6,
+        3.112949343631044e-6,
+        0.00002514909101591911,
+        0.00001734476609271951,
+        -0.000013882496205042116,
+        -7.167188869061647e-6,
+        0.00001790117130440194,
+        -4.3851646296388935e-6,
+        -9.349093375021766e-7,
+        -8.112259820336476e-6,
+        5.336457888915902e-6,
+        3.3939111290237634e-6,
+        -4.019582320324844e-6,
+        -0.00001823278762458358,
+        1.6200752952499897e-6,
+        2.1453445242514135e-6,
+        0.000018565873688203283,
+        0.00004305976108298637,
+        -0.000010276943612552714,
+        -5.3131607273826376e-6,
+        0.000013141003364580683,
+        -3.3397307106497465e-6,
+        -7.699964044149965e-7,
+        -5.316836904967204e-6,
+        -0.00004110661393497139,
+        2.852339093806222e-6,
+        -2.3958800738910213e-6,
+        -0.000013731451872445177,
+        1.5232127452691202e-6,
+        1.386316398566123e-6,
+        0.000013007000234210864,
+        0.00003866312908940017,
+        -7.723581802565604e-6,
+        -3.880812528223032e-6,
+        9.895188668451738e-6,
+        -2.4477972146996763e-6,
+        -5.607461162071559e-7,
+        -3.6636674849432893e-6,
+        0.000010404156455479097,
+        2.0179843431833433e-6,
+        -2.3253292056324426e-6,
+        -0.000010352971003158018,
+        1.0408912203274667e-6,
+        1.1318675205984619e-6,
+        9.978723028325476e-6,
+        -9.606495950720273e-6,
+        -5.949066235189093e-6,
+        -2.8392857984727016e-6,
+        7.4962795224564616e-6,
+        -1.9217382032366004e-6,
+        -4.780694098371896e-7,
+        -3.34773517352005e-6,
+        0.00002107612817781046,
+        1.6454457636427833e-6,
+        -1.817815473259543e-6,
+        -8.123145562421996e-6,
+        7.675727147216094e-7,
+        9.284493671657401e-7,
+        7.868961802159902e-6,
+        -9.203896524923039e-7,
+        -4.736490154755302e-6,
+        -2.4710741399758263e-6,
+        6.142801339592552e-6,
+        -1.6350784335372737e-6,
+        -4.335802543664613e-7,
+        -1.4092432820689282e-6,
+        0.00001656184576859232,
+        1.3327525039130705e-6,
+        -1.3708133792533772e-6,
+        -6.531353847094579e-6,
+        5.90599142924475e-7,
+        6.878227623019484e-7,
+        6.5318749875586946e-6,
+        -4.229010301060043e-6,
+        -4.000812168669654e-6,
+        -2.2653048290521838e-6,
+        5.042587417847244e-6,
+        -1.4821022205069312e-6,
+        -3.451863506143127e-7,
+        -2.389324890827993e-6,
+        8.16222745925188e-6,
+        1.1428018069636892e-6,
+        -1.1332826943544205e-6,
+        -5.504417913471116e-6,
+        5.024350002713618e-7,
+        5.467733217301429e-7,
+        5.279353445075685e-6,
+        -0.0000211214974115137,
+        -3.1678093819209607e-6,
+        -1.8488366322344518e-6,
+        4.2470492189750075e-6,
+        -9.783543646335602e-7,
+        -1.83213785476255e-7,
+        -1.4789864053454949e-6,
+        0.000013087515981169418,
+        7.242887818392774e-7,
+        -1.0046998113466543e-6,
+        -4.340507530287141e-6,
+        3.3989198300332646e-7,
+        5.470611199598352e-7,
+        4.507525773078669e-6,
+        -0.000016270019841613248,
+        -2.829020104400115e-6,
+        -1.296591335631092e-6,
+        3.456672175161657e-6,
+        -9.017520028464787e-7,
+        -1.4440175277741218e-7,
+        -1.4689791214550496e-6,
+        0.000021527563148993067,
+        5.846809472132009e-7,
+        -5.933143256697804e-7,
+        -3.612318550949567e-6,
+        3.5949764765064174e-7,
+        4.328513796281186e-7,
+        3.6886276575387456e-6,
+        -0.000021549723896896467,
+        -2.1144403490325203e-6,
+        -1.4159754755382892e-6,
+        2.68504163614125e-6,
+        -7.679741429456044e-7,
+        -5.776412592695124e-8,
+        -1.192243757941469e-6,
+        0.00003148229370708577,
+        5.478635785038932e-7,
+        -6.972362598389736e-7,
+        -3.211202283637249e-6,
+        2.2451455095051642e-7,
+        3.183014598562295e-7,
+        2.939946398328175e-6,
+        4.565895324049052e-6,
+        -1.7405694734407007e-6,
+        -9.741680742081371e-7,
+        2.2756712496629916e-6,
+        -5.214625389271532e-7,
+        -2.7989250384052866e-7,
+        -8.445824164482474e-7,
+        0.00003105050200247206,
+        5.799903988190636e-7,
+        -7.536694397458632e-7,
+        -2.5344147616124246e-6,
+        2.630972346651106e-7,
+        2.9163163617340615e-7,
+        2.3315690214076312e-6,
+        0.000028919430405949242,
+        -1.4791980902373325e-6,
+        -7.087014637363609e-7,
+        2.1661373921233462e-6,
+        -5.724799052586604e-7,
+        -3.0745661661057966e-7,
+        -5.235916091805848e-7,
+        0.000011437900866440032,
+        3.404710184895521e-7,
+        -4.5958552163938293e-7,
+        -2.1246926280582557e-6,
+        2.5418461291337735e-7,
+        3.871034266467177e-7,
+        1.9572280507418327e-6,
+        0.00003196126999682747,
+        -1.3085544878777e-6,
+        -6.04459614805819e-7,
+        1.575823830535228e-6,
+        -4.1182735799338843e-7,
+        -1.1744142369707333e-7,
+        -1.3322799077286618e-6,
+        2.7163066533830715e-6,
+        3.6782887491426663e-7,
+        -2.6708795530794305e-7,
+        -1.72060231307114e-6,
+        1.8409495794458053e-7,
+        3.7163931665418204e-7,
+        1.8107581354342983e-6,
+        0.000013270215276861563,
+        -1.1510371678014053e-6,
+        -6.630299367316184e-7,
+        1.070535518010729e-6,
+        -3.826233978543314e-7,
+        1.1823525625231923e-7,
+        -1.049707634592778e-6,
+        -2.8137444587628124e-6,
+        2.6974137767865614e-7,
+        -4.0497872078049113e-7,
+        -1.4479496712738182e-6,
+        3.8844518712721765e-8,
+        2.0770157149740953e-8,
+        1.7100152263083146e-6,
+        -0.00002849056909326464,
+        -9.183266342915886e-7,
+        -3.5311214219291287e-7,
+        1.1368990726623451e-6,
+        -3.410139868265105e-7,
+        -2.7979771743957826e-7,
+        -9.036052119881788e-7,
+        -8.269290447060484e-6,
+        3.0624855185124034e-7,
+        -1.1787745535229988e-7,
+        -1.2770473176715313e-6,
+        1.6275393477371836e-7,
+        3.3039643199117563e-7,
+        1.3103665423841449e-6,
+        -0.00001745441295497585,
+        -6.018551061970356e-7,
+        -2.794714930587361e-7,
+        9.792224773264024e-7,
+        -2.040594893060188e-7,
+        -3.4366582468692286e-8,
+        -4.681474479184544e-7,
+        -3.3097076084231958e-6,
+        1.669202376319845e-8,
+        -2.7714148131963157e-7,
+        -9.64512764767278e-7,
+        1.1432685909085194e-7,
+        1.4834664341378812e-7,
+        1.1487189794934238e-6,
+        -0.000010153660696232691,
+        -6.140094228612725e-7,
+        -5.253555173112545e-7,
+        5.710655841539847e-7,
+        -1.404835359153367e-7,
+        4.7323091223461233e-8,
+        -5.46001842849364e-7,
+        -1.2161887070760713e-6,
+        1.2690516371094418e-7,
+        -1.3310798863130913e-7,
+        -7.572793379040377e-7,
+        1.7468086355165724e-7,
+        3.3051062331423964e-8,
+        7.131747565836122e-7,
+        -0.00003459135041339323,
+        -4.08122900807939e-7,
+        -1.8256875478073198e-7,
+        5.756255063715798e-7,
+        -1.6261435575870564e-7,
+        -2.021908862559485e-8,
+        -4.1869603251143417e-7,
+        -1.589809244251228e-6,
+        1.1844038283470582e-7,
+        -7.242162070042468e-8,
+        -4.0011059354583267e-7,
+        -2.4760929306921753e-8,
+        -1.3480507732310798e-7,
+        1.0380339290350094e-6,
+        1.6824413933136384e-6,
+        -4.5796767267347605e-7,
+        -2.0989804738746898e-7,
+        4.763822971654008e-7,
+        -1.7666546625605406e-7,
+        -2.998827852707109e-8,
+        3.5782920804194873e-7,
+        -0.00005497613165061921,
+        -4.8836572119626e-8,
+        -2.8484058134381485e-7,
+        -2.0204961970193835e-7,
+        -1.9280198770843526e-8,
+        8.62186144701127e-9,
+        3.8213460129554733e-7,
+        1.619790509721497e-6,
+        -1.449097339900618e-7,
+        -1.7092831683385157e-7,
+        3.423832595217391e-7,
+        -9.912067611139719e-8,
+        5.862289853553193e-8,
+        -7.740770513464668e-8,
+        -7.812944204488304e-6,
+        1.33551338876714e-7,
+        1.7766551252407226e-7,
+        -2.63828656699161e-8,
+        -5.126612023786947e-9,
+        -4.134697562108158e-8,
+        8.136574791706153e-8,
+        -4.875783361057984e-6
+      ],
+      "std": [
+        0.054471205919981,
+        0.07741593569517136,
+        0.06635452806949615,
+        0.02219589799642563,
+        0.03713744506239891,
+        0.2155904769897461,
+        0.44884487986564636,
+        0.022129571065306664,
+        0.030908629298210144,
+        0.027230359613895416,
+        0.009623671881854534,
+        0.015185607597231865,
+        0.08782459795475006,
+        0.11000800132751465,
+        0.0054918802343308926,
+        0.007706818170845509,
+        0.007092009298503399,
+        0.004339421167969704,
+        0.006056886166334152,
+        0.025186194106936455,
+        0.0549297109246254,
+        0.0023341968189924955,
+        0.003104041563346982,
+        0.0028293700888752937,
+        0.002068394096568227,
+        0.0026343404315412045,
+        0.01149396225810051,
+        0.03640395402908325,
+        0.0011078902753069997,
+        0.0015103736659511924,
+        0.0014142218278720975,
+        0.0010499481577426195,
+        0.0013543356908485293,
+        0.005356063600629568,
+        0.027538813650608063,
+        0.0007573548937216401,
+        0.001025661244057119,
+        0.0009198821499012411,
+        0.0005544671439565718,
+        0.0007547291461378336,
+        0.0035150113981217146,
+        0.02213539369404316,
+        0.000455506524303928,
+        0.0006300596869550645,
+        0.0005910721374675632,
+        0.00034599038190208375,
+        0.0004919870407320559,
+        0.002043767599388957,
+        0.018490547314286232,
+        0.00037283002166077495,
+        0.0005109719350002706,
+        0.00045568859786726534,
+        0.00023278511071112007,
+        0.00032812528661452234,
+        0.0016570856096222997,
+        0.015843955799937248,
+        0.0002481868432369083,
+        0.0003480722662061453,
+        0.000325335975503549,
+        0.00017280328029301018,
+        0.0002512190258130431,
+        0.0010802638716995716,
+        0.01408928632736206,
+        0.00022223185806069523,
+        0.00030612919363193214,
+        0.0002718937466852367,
+        0.00012990496179554611,
+        0.00018422518041916192,
+        0.000973312184214592,
+        0.012508493848145008,
+        0.00015761872055009007,
+        0.00022044251090846956,
+        0.0002061627892544493,
+        0.0001057182380463928,
+        0.00015497820277232677,
+        0.0006764288409613073,
+        0.011296260170638561,
+        0.00014791855937801301,
+        0.00020399357890710235,
+        0.00018064411415252835,
+        0.00008491610060445964,
+        0.0001210872214869596,
+        0.0006432573427446187,
+        0.01023485790938139,
+        0.00010906582610914484,
+        0.00015326884749811143,
+        0.000142823948408477,
+        0.00007245635788422078,
+        0.00010654709330992773,
+        0.00046615453902632,
+        0.009492816403508186,
+        0.00010573185136308894,
+        0.0001460392086300999,
+        0.0001292345259571448,
+        0.00006049493094906211,
+        0.00008650264499010518,
+        0.00045887212036177516,
+        0.008779959753155708,
+        0.00007973915489856154,
+        0.00011203208850929514,
+        0.00010434568685013801,
+        0.00005302290082909167,
+        0.00007767079659970477,
+        0.0003416067629586905,
+        0.00814682524651289,
+        0.00007915469905128703,
+        0.00010886993550229818,
+        0.0000966577063081786,
+        0.000045318782213144004,
+        0.00006455113179981709,
+        0.0003425226896069944,
+        0.007630111649632454,
+        0.0000611644281889312,
+        0.00008592260564910248,
+        0.00007935166649986058,
+        0.000040576029277872294,
+        0.00005935502122156322,
+        0.00026016749325208366,
+        0.007179272826761007,
+        0.00006150962872197852,
+        0.00008518873801222071,
+        0.00007461248605977744,
+        0.0000355022020812612,
+        0.0000502788316225633,
+        0.00026474526384845376,
+        0.006768743507564068,
+        0.000048591213271720335,
+        0.00006827081961091608,
+        0.00006262768874876201,
+        0.000032297353754984215,
+        0.000046926012146286666,
+        0.00020619359565898776,
+        0.006469320971518755,
+        0.00004949716094415635,
+        0.00006947917427169159,
+        0.00005963100556982681,
+        0.000028536098398035392,
+        0.000040751590859144926,
+        0.00021295368787832558,
+        0.006215229630470276,
+        0.00003966964141000062,
+        0.00005751234493800439,
+        0.00005083374344394542,
+        0.00002690069413802121,
+        0.00003854740134556778,
+        0.00016899738693609834,
+        0.005915363319218159,
+        0.00004071096918778494,
+        0.00005744508234784007,
+        0.000049411490181228146,
+        0.000024393977582803927,
+        0.000034359978599241003,
+        0.00017412411398254335,
+        0.005679433699697256,
+        0.000034098236938007176,
+        0.00004947670822730288,
+        0.0000428831554017961,
+        0.000023318862076848745,
+        0.0000333583120664116,
+        0.00014037704386282712,
+        0.005502832122147083,
+        0.000035476441553328186,
+        0.00005029448220739141,
+        0.000041998053347924724,
+        0.000021735353584517725,
+        0.00003051272506127134,
+        0.00014704464410897344,
+        0.005227266810834408,
+        0.000031422849133377895,
+        0.0000454086039098911,
+        0.000037437119317473844,
+        0.000021104986444697715,
+        0.0000290283169306349,
+        0.00012362531560938805,
+        0.005091419909149408,
+        0.00003195118915755302,
+        0.00004717931369668804,
+        0.00003730821481440216,
+        0.000018998216546606272,
+        0.00002651557770150248,
+        0.0001294043322559446,
+        0.004971646703779697,
+        0.00002650190617714543,
+        0.000040163333324017,
+        0.00003248209395678714,
+        0.000018027179976343177,
+        0.00002582307206466794,
+        0.00010801247117342427,
+        0.004754771012812853,
+        0.00002661882717802655,
+        0.00003864263999275863,
+        0.00003127731179120019,
+        0.000016530248103663325,
+        0.000023721948309685104,
+        0.00010995724005624652,
+        0.004578577820211649,
+        0.000022862250261823647,
+        0.000033582466130610555,
+        0.000027823365599033423,
+        0.000015542551409453154,
+        0.000022585352780879475,
+        0.00009167923417408019,
+        0.004508121404796839,
+        0.00002280540320498403,
+        0.000033627373341005296,
+        0.00002707460953388363,
+        0.000014449000445893034,
+        0.00002091616806865204,
+        0.0000951679321588017,
+        0.004384348168969154,
+        0.00002045510154857766,
+        0.00002992084773723036,
+        0.000024755014237598516,
+        0.000014510045730276033,
+        0.000020438790670596063,
+        0.00008195763803087175,
+        0.004295016173273325,
+        0.000021367646695580333,
+        0.00003022674354724586,
+        0.000024624087018310092,
+        0.000013480367670126725,
+        0.000019035556761082262,
+        0.0000849158750497736,
+        0.004229281097650528,
+        0.000018941680536954664,
+        0.000027560265152715147,
+        0.000022634478227701038,
+        0.000013048511391389184,
+        0.00001906735087686684,
+        0.00007548843132099137,
+        0.004119527991861105,
+        0.00002021879299718421,
+        0.00002808637873386033,
+        0.000022752908989787102,
+        0.000012890232028439641,
+        0.000019045859517063946,
+        0.0000775336375227198,
+        0.004028849303722382,
+        0.00001860819975263439,
+        0.000026566647647996433,
+        0.000021319850930012763,
+        0.000012895960026071407,
+        0.00001921564035001211,
+        0.0000719531744834967,
+        0.003980362322181463,
+        0.000019542083464330062,
+        0.000027795384085038677,
+        0.000021858841137145646,
+        0.000013172734725230839,
+        0.000019754204913624562,
+        0.00007459476182702929,
+        0.003902778960764408,
+        0.00002043036147370003,
+        0.00003195027602487244,
+        0.000024560755264246836,
+        0.000014010387531016022,
+        0.00002065290027530864,
+        0.00007676216046093032,
+        0.0038293309044092894,
+        0.000023211092411656864,
+        0.00003743018169188872,
+        0.000025842035029199906,
+        0.000015010549759608693,
+        0.000020894838598906063,
+        0.00008554229862056673,
+        0.0037848083302378654,
+        0.00002203849180659745,
+        0.00003869964712066576,
+        0.00002533109909563791,
+        0.000013811703865940217,
+        0.000020183759261271916,
+        0.0000796892709331587,
+        0.0037312512286007404,
+        0.000020139470507274382,
+        0.00003185970854246989,
+        0.00002326746835024096,
+        0.000014411976735573262,
+        0.000020371249775053002,
+        0.00007360208837781101,
+        0.0036642765626311302,
+        0.000017049775124178268,
+        0.000027688640329870395,
+        0.00001989392148971092,
+        0.000012698577847913839,
+        0.000017972832210944034,
+        0.00006560993642779067,
+        0.0036550285294651985,
+        0.000016028958270908333,
+        0.000024884970116545446,
+        0.00001818954842747189,
+        0.000011278674719505943,
+        0.000015812189303687774,
+        0.00006195339665282518,
+        0.0036316923797130585,
+        0.00001560178498039022,
+        0.000024082159143290482,
+        0.00001761745625117328,
+        0.00001089257057174109,
+        0.000015621577404090203,
+        0.00005938451431575231,
+        0.003624820150434971,
+        0.000014879788068355992,
+        0.00002372561357333325,
+        0.00001737473940011114,
+        0.00001018085913528921,
+        0.000015368110325653106,
+        0.00005791655712528154,
+        0.0035625966265797615,
+        0.000014492210539174266,
+        0.000023905164198367856,
+        0.000016788300854386762,
+        0.00001016644364426611,
+        0.000014378600099007599,
+        0.00005657917427015491,
+        0.0035458113998174667,
+        0.000014646487215941306,
+        0.000023379501726594754,
+        0.000016420719475718215,
+        0.000010098652637680061,
+        0.000014440963241213467,
+        0.00005597082417807542,
+        0.0035429371055215597,
+        0.000014022874893271364,
+        0.00002083050821966026,
+        0.000015985659047146328,
+        0.00001016553051158553,
+        0.000013916131138103083,
+        0.000056019474868662655,
+        0.0034998103510588408,
+        0.000014245401871448848,
+        0.000022099453417467885,
+        0.00001657450047787279,
+        0.000010621713045111392,
+        0.000014651923265773803,
+        0.00005747186514781788,
+        0.0034743566066026688,
+        0.000014478311641141772,
+        0.00002349081842112355,
+        0.000017406200640834868,
+        0.000010810344974743202,
+        0.000014864897821098566,
+        0.00005688491728506051,
+        0.0034770709462463856,
+        0.000016515868992428295,
+        0.00002587365270301234,
+        0.000018931375961983576,
+        0.000010814174856932368,
+        0.000016132373275468126,
+        0.00006046127964509651,
+        0.003478288883343339
+      ],
+      "q01": [
+        -0.15909142580032348,
+        -0.23591104531884194,
+        -0.1173719973474741,
+        -0.0725411132991314,
+        -0.12759929532110692,
+        -0.6792203711509704,
+        0.0,
+        -0.06588815684318543,
+        -0.1043403072744608,
+        -0.07569241938591004,
+        -0.026123490297794344,
+        -0.04355562012195587,
+        -0.26860033273696904,
+        -0.31594268302917483,
+        -0.017998227071762084,
+        -0.025448361220955847,
+        -0.021794256627559662,
+        -0.014516740383952856,
+        -0.017180888187885286,
+        -0.08126688668131828,
+        -0.15649057692885399,
+        -0.006211889632977545,
+        -0.009136751361563802,
+        -0.00862356810811907,
+        -0.00636542599722743,
+        -0.007391496415808798,
+        -0.035477512252330784,
+        -0.10543232785761356,
+        -0.003451917093619704,
+        -0.004779039949178695,
+        -0.00400189593732357,
+        -0.003541165339201688,
+        -0.003899161383137107,
+        -0.016611159443855286,
+        -0.07849292940199376,
+        -0.001959954298567027,
+        -0.0028813693393021818,
+        -0.0027899963006842885,
+        -0.0015735435530543325,
+        -0.002058688412979245,
+        -0.010417635813355447,
+        -0.0639662309139967,
+        -0.0013655057244002819,
+        -0.0019750579357147215,
+        -0.0016051418443210422,
+        -0.0011298993272706869,
+        -0.0014582404736429451,
+        -0.006558569058775902,
+        -0.05251251416504383,
+        -0.0009643727889051661,
+        -0.0014315907768905163,
+        -0.001390606227517128,
+        -0.0006525630556046963,
+        -0.0008888287223875523,
+        -0.004895182948186994,
+        -0.0455008171916008,
+        -0.0007486765272915363,
+        -0.0010981693975627422,
+        -0.000882065705396235,
+        -0.0005725251567782834,
+        -0.0007578355811536312,
+        -0.0034509299120865766,
+        -0.04015725544691086,
+        -0.0005749710650416091,
+        -0.0008555282396264373,
+        -0.0008345671106129884,
+        -0.0003548953181132674,
+        -0.0004969418092630804,
+        -0.0028254041438922285,
+        -0.035569155138731004,
+        -0.0004705392210977152,
+        -0.0006840464788489043,
+        -0.0005612756710499526,
+        -0.00035702317333780227,
+        -0.0004648537166416645,
+        -0.002128881996870041,
+        -0.03083838468044997,
+        -0.00038048800966935235,
+        -0.000566111233481206,
+        -0.0005538229705183766,
+        -0.00023378437627106905,
+        -0.00032636490603908896,
+        -0.0018706658044829966,
+        -0.02931403119564057,
+        -0.00032479510342236606,
+        -0.000475912622641772,
+        -0.00038455819835653526,
+        -0.00024285348085686562,
+        -0.0003135723091545515,
+        -0.001452100250311196,
+        -0.026792802512645722,
+        -0.00027214939426630733,
+        -0.00040756029058247796,
+        -0.0003960524151916616,
+        -0.00016536210232879968,
+        -0.00023339646961539982,
+        -0.0013170160830486568,
+        -0.02499642580747604,
+        -0.00024108570310054343,
+        -0.000348888192162849,
+        -0.0002832006874959916,
+        -0.00017626483961939813,
+        -0.00023037131156306717,
+        -0.0010707858602516352,
+        -0.023444418999552727,
+        -0.00020677300081588327,
+        -0.0003006568572251126,
+        -0.0002946901318966411,
+        -0.00012258605094393713,
+        -0.00017606193688698117,
+        -0.0009890523520298302,
+        -0.02203794540464878,
+        -0.0001841578755469527,
+        -0.00026803012369200585,
+        -0.00021574946893379094,
+        -0.00013577964630676436,
+        -0.0001734850220615044,
+        -0.0008068450814578683,
+        -0.020441363253444433,
+        -0.00016045538827893325,
+        -0.000241462552268058,
+        -0.00022808200246654452,
+        -0.00009496491550235078,
+        -0.00013633702206425368,
+        -0.0007666170152835548,
+        -0.019491385824978353,
+        -0.0001477667265571654,
+        -0.00021141915648477145,
+        -0.0001735861329594627,
+        -0.00010635797707363962,
+        -0.00013736910806037486,
+        -0.0006427704328671098,
+        -0.018371846283972263,
+        -0.0001293026403174736,
+        -0.00019473347970051693,
+        -0.0001822685490187723,
+        -0.00007863951541366989,
+        -0.00010972291373182087,
+        -0.0006055113250389692,
+        -0.017673968444019555,
+        -0.00011848664400167762,
+        -0.00017848168123746295,
+        -0.00014009435730986296,
+        -0.0000859230127884075,
+        -0.00011260414165444673,
+        -0.0005137789773289115,
+        -0.016190811094641686,
+        -0.00010690572904422879,
+        -0.0001638943360070698,
+        -0.0001501999743981287,
+        -0.00006396331703872426,
+        -0.00009358972666086629,
+        -0.0005007586104329676,
+        -0.01619275164604187,
+        -0.00010315810496686024,
+        -0.00015018478848505767,
+        -0.00011576190601917915,
+        -0.00007506265956908466,
+        -0.00009504533028230073,
+        -0.0004348065205849708,
+        -0.01565815359354019,
+        -0.00009185501914471398,
+        -0.0001392430538544431,
+        -0.00012555517936125394,
+        -0.00005638334401883189,
+        -0.00008002441792050375,
+        -0.000424562167050317,
+        -0.015019814673811197,
+        -0.00009095547425094991,
+        -0.00013837829248514026,
+        -0.00010207304285722785,
+        -0.00006334332783007992,
+        -0.00008174893613904723,
+        -0.0003743791728978976,
+        -0.014377124969661235,
+        -0.00008569652531296014,
+        -0.00013066508993506436,
+        -0.00010974786025471983,
+        -0.000049689067096915106,
+        -0.00007135935623664408,
+        -0.0003742531020194291,
+        -0.014187388128042218,
+        -0.0000794089021859691,
+        -0.00011592403496615584,
+        -0.0000893614823697135,
+        -0.0000560631047235802,
+        -0.0000719853327609599,
+        -0.0003315736224874853,
+        -0.013546264378726483,
+        -0.0000717077999375761,
+        -0.00010777361229993398,
+        -0.0000922710225568153,
+        -0.00004497638295288198,
+        -0.00006191108026541768,
+        -0.0003081054835114628,
+        -0.013243352345749734,
+        -0.00006864526765421036,
+        -0.0001027144630090333,
+        -0.00008054810634930621,
+        -0.000049568866193294524,
+        -0.00006090367061551658,
+        -0.00028183929068036377,
+        -0.012778499507904054,
+        -0.00006117501191329209,
+        -0.00009796037861378861,
+        -0.00008128996908781118,
+        -0.00004006202814634888,
+        -0.00005725241685286168,
+        -0.0002700211696093902,
+        -0.012565325374901296,
+        -0.00006034678600262851,
+        -0.0000907877254066989,
+        -0.00007010409524082204,
+        -0.000041811013757251194,
+        -0.0000556492382194847,
+        -0.0002493678867351264,
+        -0.011774876400828364,
+        -0.00005579026520717887,
+        -0.00008717787461355332,
+        -0.00007443824684014546,
+        -0.0000356287220492959,
+        -0.00005372272633831019,
+        -0.00024090840241406124,
+        -0.011997990143299102,
+        -0.000058186027396004656,
+        -0.00008279621332185345,
+        -0.00006459413380362097,
+        -0.00003831381897907702,
+        -0.00005259611281217076,
+        -0.00022310287454165522,
+        -0.011667696542292834,
+        -0.00005284995273686948,
+        -0.00008384203007444737,
+        -0.00006744749160716312,
+        -0.000032252046390203774,
+        -0.0000481185418262612,
+        -0.00021629847201984375,
+        -0.011543062198162084,
+        -0.00005847320245811716,
+        -0.00008175599463284015,
+        -0.00006548594534397124,
+        -0.000038102848210837704,
+        -0.0000492544468608685,
+        -0.00021983185317367318,
+        -0.011402166564017536,
+        -0.00005387163197156041,
+        -0.0000825874566333369,
+        -0.00006687336476170457,
+        -0.000033979644608916735,
+        -0.00005101760845864191,
+        -0.00020870170197449626,
+        -0.011224610126763582,
+        -0.00006089058108045721,
+        -0.00008930324914399536,
+        -0.00007119371397420766,
+        -0.00003768286973936482,
+        -0.000053939796844497314,
+        -0.0002299774368293584,
+        -0.010891138190776108,
+        -0.000060360829543788016,
+        -0.00009397184122353786,
+        -0.00007481435872614382,
+        -0.00003693910902366043,
+        -0.00005306754379416818,
+        -0.0002138554411940277,
+        -0.01082267729640007,
+        -0.00005871810513781385,
+        -0.00008991035101935281,
+        -0.00007033640673034824,
+        -0.000036662435624748536,
+        -0.00005430486062541601,
+        -0.0002200629871338608,
+        -0.010579705061763527,
+        -0.000053560086898505666,
+        -0.00008582322522997861,
+        -0.00007200759416446087,
+        -0.000034787034674081956,
+        -0.000048553584120236346,
+        -0.0002130190141499044,
+        -0.010566252829506993,
+        -0.00005062428400851786,
+        -0.00007914942941861225,
+        -0.000059331482945708576,
+        -0.00003430225813062863,
+        -0.00004731270743068308,
+        -0.00019290218278765676,
+        -0.010022874112427235,
+        -0.000047412463498767456,
+        -0.00007506768405437467,
+        -0.00005783305459190161,
+        -0.00002997652450576426,
+        -0.00004301143199554646,
+        -0.00018155146474018715,
+        -0.010345511507242918,
+        -0.00004604501994326713,
+        -0.00007326857978478065,
+        -0.00005605542755220084,
+        -0.000028739307308569553,
+        -0.00004117103526368737,
+        -0.0001768201191909612,
+        -0.010309934300184249,
+        -0.00004275882025249304,
+        -0.00006731156632304187,
+        -0.00005409912285394967,
+        -0.000026329884584993103,
+        -0.00004337953124195339,
+        -0.00016598801976069823,
+        -0.01018075596317649,
+        -0.000043296541273593885,
+        -0.00006625080597586939,
+        -0.00005305007540155199,
+        -0.000028063857986126094,
+        -0.000040029496920760744,
+        -0.00017517980523407465,
+        -0.010162209048867225,
+        -0.0000426195073465351,
+        -0.00006396621902240442,
+        -0.000051819634152343525,
+        -0.000029033373587299122,
+        -0.00004320319109829139,
+        -0.00016554977782070636,
+        -0.010125353442132473,
+        -0.000045235728612169645,
+        -0.00006515037268400192,
+        -0.00004881905938964337,
+        -0.000028501104679889977,
+        -0.00004041612264700232,
+        -0.00016903663412667815,
+        -0.010062240642309189,
+        -0.000043975374847650503,
+        -0.00006714530037716033,
+        -0.00005093284538015727,
+        -0.00002801080799545163,
+        -0.00004157804111018775,
+        -0.00016746679800562535,
+        -0.009975232541561126,
+        -0.000045126923860516424,
+        -0.00007225614520721142,
+        -0.00005288218818604951,
+        -0.000029170855134725583,
+        -0.000040984369814395916,
+        -0.0001663898449391129,
+        -0.009851353891193867,
+        -0.00004605035483837126,
+        -0.00007112415134906766,
+        -0.0000582737371325493,
+        -0.000028596166521310774,
+        -0.000044461411982774806,
+        -0.0001793744623661041,
+        -0.009932486386597156
+      ],
+      "q99": [
+        0.16212226042747496,
+        0.25229084351658815,
+        0.18486946655809877,
+        0.061344154804944984,
+        0.10886132830679413,
+        0.6657196632862092,
+        0.9998,
+        0.06576045138835906,
+        0.09603128664791583,
+        0.045872775423526774,
+        0.03236173896193503,
+        0.05487722767591477,
+        0.2764705369472503,
+        0.3158153381347656,
+        0.014891946853697303,
+        0.023800758841633796,
+        0.018585946798324586,
+        0.012693745650351043,
+        0.021670551879704,
+        0.08171248322725297,
+        0.15637542012631894,
+        0.0069066042726859435,
+        0.009335029612295327,
+        0.0047039436949416966,
+        0.006996506480127576,
+        0.008702572771534321,
+        0.036967494320869446,
+        0.1053792189329863,
+        0.002857540221139789,
+        0.004296822899580001,
+        0.003630450583249331,
+        0.003087930056080223,
+        0.004339818637073041,
+        0.01760330287218094,
+        0.07830542102754112,
+        0.0021181412500329315,
+        0.0029414672447368503,
+        0.0015112007031682874,
+        0.0019269651249051098,
+        0.0024494722973555337,
+        0.01144827667623758,
+        0.06380807581543924,
+        0.0011728783883154393,
+        0.001813931218627841,
+        0.0014501392262987797,
+        0.0009983487627469,
+        0.0015205995496362446,
+        0.006439514055848122,
+        0.05315425983667374,
+        0.0010328240715665742,
+        0.0014663059995509691,
+        0.0007404281431809068,
+        0.0008001560054719447,
+        0.001095746040716767,
+        0.0052222565246745935,
+        0.04538371631056071,
+        0.0006383903683163227,
+        0.0009848102391697465,
+        0.000784420038573444,
+        0.0004887675133766608,
+        0.0007893593505024911,
+        0.0033699059166945513,
+        0.039476715770363824,
+        0.0006163579971762373,
+        0.0008800703473389147,
+        0.00044086696542799475,
+        0.000447366009419784,
+        0.0006247742644743994,
+        0.0030928143300116076,
+        0.035530846929550164,
+        0.0004072732983855529,
+        0.0006279987238347533,
+        0.0004935565162450075,
+        0.0003003585597965866,
+        0.0004761313281953334,
+        0.002095991748571396,
+        0.030764527727663513,
+        0.00040901676943758494,
+        0.0005783375784289091,
+        0.00029091126989806075,
+        0.0002924530396237969,
+        0.0004053453548112883,
+        0.0020267377858981485,
+        0.02924695993065833,
+        0.0002835550721851177,
+        0.0004330007825046778,
+        0.0003438553759129719,
+        0.00021280083870515228,
+        0.0003332204633043148,
+        0.0014287737663835282,
+        0.02666576117277146,
+        0.00029395890273153794,
+        0.00041647118404507634,
+        0.0002144097874115686,
+        0.00020652228510007256,
+        0.0002845425325096586,
+        0.0014505576011724767,
+        0.024956510066986085,
+        0.00020434754303423695,
+        0.00031225532339885847,
+        0.0002542992958333343,
+        0.0001512215134222061,
+        0.00024444468863075595,
+        0.0010491569862701002,
+        0.023069663181155914,
+        0.00022068456518463788,
+        0.00031388382902368915,
+        0.00016011388470651587,
+        0.00015496468472993004,
+        0.00021093179946765303,
+        0.001086882085958496,
+        0.0219863745495677,
+        0.00016183016332215621,
+        0.00024126361515372995,
+        0.0001919538950081915,
+        0.00011462044673971831,
+        0.00018409015340730545,
+        0.0008034803080838172,
+        0.020384925924241543,
+        0.0001725941591721494,
+        0.00024417606623610485,
+        0.00012369809069205076,
+        0.00012043028190964833,
+        0.00016321412267861885,
+        0.0008335524674970658,
+        0.019483701366186136,
+        0.00012494574459269637,
+        0.0001922543402411975,
+        0.00015176694963593034,
+        0.00008988425261923113,
+        0.00014705147650092838,
+        0.0006307619461789725,
+        0.01831897035390139,
+        0.00013962995521724217,
+        0.00019706185359973463,
+        0.00010126548151019962,
+        0.0000950566966785118,
+        0.00013232208000263205,
+        0.000660218992456794,
+        0.01764387315139175,
+        0.00010247177979908883,
+        0.00015943087645573536,
+        0.00012344940464245158,
+        0.00007527396054356363,
+        0.00011956113649066536,
+        0.0005160252890083939,
+        0.016168088349699967,
+        0.00011438966927817092,
+        0.00016185711201978847,
+        0.00008490921122720466,
+        0.00007747934755752788,
+        0.00010678794741397725,
+        0.0005481199477799237,
+        0.01615487493276596,
+        0.00008468688050634237,
+        0.00013357372635509814,
+        0.00010555801286245698,
+        0.00006330196186900142,
+        0.000101998839713633,
+        0.00042467831117101035,
+        0.015395162180066112,
+        0.00009998356355354183,
+        0.00014026010758243506,
+        0.00007274458486353978,
+        0.00006777426579501484,
+        0.00009303196137771012,
+        0.0004640237890882414,
+        0.01499328883215785,
+        0.00007716326334048071,
+        0.00011421011368511244,
+        0.00009314750883495434,
+        0.00005738810400944202,
+        0.00008783590141683817,
+        0.0003717806453816595,
+        0.014554040935635573,
+        0.00008682058225385845,
+        0.00012317761406302453,
+        0.00007470068163238468,
+        0.00005833794826176018,
+        0.0000807305663009174,
+        0.0003966953692026436,
+        0.014136480166763069,
+        0.00006818514921469611,
+        0.00010206100093200813,
+        0.00008256861115805807,
+        0.000049349277792498495,
+        0.00007737125915009528,
+        0.00031947106840088976,
+        0.013474849896132943,
+        0.00007525306886527682,
+        0.00010829368010163308,
+        0.00006185477034305223,
+        0.000050571078446228043,
+        0.00007023279787972571,
+        0.0003371094720438124,
+        0.013198494754731657,
+        0.000059138852264732124,
+        0.00008794275637483227,
+        0.00007062362715369087,
+        0.00004125927500426771,
+        0.00006884480739245192,
+        0.0002681864194571972,
+        0.012739881089329719,
+        0.00006564935145433995,
+        0.00009313276901375502,
+        0.00005367935128742829,
+        0.00004326853083912283,
+        0.000059486858663149135,
+        0.000290332669764757,
+        0.012545088732242582,
+        0.00005332418816396964,
+        0.0000798318615881726,
+        0.00006209178492426877,
+        0.00003812435691943395,
+        0.00005920125464908778,
+        0.00024440587060526003,
+        0.011752966091036793,
+        0.00006044615641003474,
+        0.00008350273473188281,
+        0.00005087533958721905,
+        0.00004026508983224629,
+        0.00005588843687437471,
+        0.0002512834585038945,
+        0.01198878407478332,
+        0.00004965451551834123,
+        0.00007573206002125512,
+        0.00006231120838783678,
+        0.00003364615770988169,
+        0.00005282912561669945,
+        0.0002209204529877752,
+        0.011624232469499118,
+        0.00005783643363974989,
+        0.00008142779953777802,
+        0.00005095888577052386,
+        0.000036499108670977874,
+        0.000052707964507862936,
+        0.000231120909564197,
+        0.01148185554742813,
+        0.00004983878775965424,
+        0.00007604824067093425,
+        0.000057698829320725065,
+        0.000033903982862830136,
+        0.000054097583203110825,
+        0.00020294103957712662,
+        0.01119744649901986,
+        0.00005782577866921202,
+        0.00007854039029916743,
+        0.0000542719262011815,
+        0.00003704023372847582,
+        0.00005206325108883899,
+        0.00022946770491544147,
+        0.011211802294850345,
+        0.0000541387221601326,
+        0.00008076134007424106,
+        0.00006221706023206938,
+        0.00003468251589220018,
+        0.000056180346361361434,
+        0.00020245690247975277,
+        0.010855805838853122,
+        0.00005655899195699014,
+        0.0000880048372084274,
+        0.00006508456927258521,
+        0.00003773696059361102,
+        0.00005296763135702342,
+        0.00022277740947902205,
+        0.010783490767329935,
+        0.00005639911381294957,
+        0.00008836033511906856,
+        0.00007074359970283693,
+        0.00003449545800685873,
+        0.00004960680939257152,
+        0.0002082344447262587,
+        0.010557891380786895,
+        0.00005552569483406838,
+        0.00008377498388290397,
+        0.0000605235195718706,
+        0.0000366905065602623,
+        0.00005407085546758027,
+        0.00021638836078345788,
+        0.010525680954381826,
+        0.00004830919423839074,
+        0.00007733448530780152,
+        0.00005978569525177588,
+        0.00003123643363360317,
+        0.00004709080595057457,
+        0.0001939378370530904,
+        0.00998856540247798,
+        0.00004645563278463669,
+        0.00007126600248739121,
+        0.00005200477748876437,
+        0.00003055265350267292,
+        0.000045497189089655854,
+        0.00018253272622823706,
+        0.01032583272829652,
+        0.000044604689953848674,
+        0.00006673079421743756,
+        0.000050660950923338505,
+        0.000027101854892680433,
+        0.000045762255322188115,
+        0.00016890384713187808,
+        0.010136223459243775,
+        0.00004455379862338299,
+        0.00006405452857725328,
+        0.00004943669138010592,
+        0.000028590407309820904,
+        0.000039900707779452174,
+        0.00017483096104115243,
+        0.010142471250146626,
+        0.0000416521602543071,
+        0.00006346322163008151,
+        0.00005030547470669263,
+        0.00002662300992524252,
+        0.00003956695257220416,
+        0.00016375841144472352,
+        0.009997803356498476,
+        0.000042872124351561063,
+        0.00006597544132964689,
+        0.00005052208236302249,
+        0.00002720812879269946,
+        0.000039130584619124446,
+        0.0001802312535233797,
+        0.010092107820510864,
+        0.00003863017375115306,
+        0.00006420832830481232,
+        0.000052780072088353346,
+        0.000026064168196171522,
+        0.000040864129550755046,
+        0.00018467701834160846,
+        0.009899575112760066,
+        0.0000431866092141718,
+        0.00006625963510014114,
+        0.00005088201165199277,
+        0.000026399783365195603,
+        0.000037794473674148364,
+        0.00017790157631970968,
+        0.00994323554635048,
+        0.00004512640612665562,
+        0.00007162263081409033,
+        0.00005516039710491889,
+        0.000026367472112178792,
+        0.00004286482669413089,
+        0.0001678395111113784,
+        0.009839660620689389,
+        0.000047105535864830025,
+        0.00007688079774379733,
+        0.00005521634444594385,
+        0.00002967622131109241,
+        0.000041797038167715075,
+        0.00017254053354263292,
+        0.009918793506920338
+      ]
+    }
+  },
+  "sample_ratio": 0.05,
+  "dataset_infos": {
+    "robotgeneralist/pp__1": {
+      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
+      "num_episodes": 237,
+      "num_frames": 99363
+    },
+    "robotgeneralist/pp__1__adversarial": {
+      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
+      "num_episodes": 64,
+      "num_frames": 17906
+    },
+    "robotgeneralist/pp__2": {
+      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
+      "num_episodes": 92,
+      "num_frames": 37458
+    },
+    "robotgeneralist/pp__2__rotations": {
+      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
+      "num_episodes": 102,
+      "num_frames": 13935
+    },
+    "robotgeneralist/pp__3__with_ft": {
+      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
+      "num_episodes": 99,
+      "num_frames": 38424
+    },
+    "robotgeneralist/pp__3__with_ft__rotations": {
+      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
+      "num_episodes": 49,
+      "num_frames": 7151
+    },
+    "robotgeneralist/pp__3__with_ft__adversarial_start": {
+      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
+      "num_episodes": 147,
+      "num_frames": 21313
+    },
+    "robotgeneralist/pp__4__with_ft": {
+      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
+      "num_episodes": 94,
+      "num_frames": 33046
+    },
+    "robotgeneralist/pp__5__with_ft__adversarial_target": {
+      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
+      "num_episodes": 48,
+      "num_frames": 10615
+    },
+    "robotgeneralist/pp__6__with_ft": {
+      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
+      "num_episodes": 379,
+      "num_frames": 123611
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-1": {
+      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
+      "num_episodes": 98,
+      "num_frames": 41410
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-2": {
+      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
+      "num_episodes": 94,
+      "num_frames": 37859
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-3": {
+      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
+      "num_episodes": 127,
+      "num_frames": 53341
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-4": {
+      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
+      "num_episodes": 102,
+      "num_frames": 38263
+    }
+  }
+}

pp__ee_fourier_k50/10000/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}

pp__ee_fourier_k50/10000/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}]]}}"}

pp__ee_fourier_k50/10000/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}}]}

pp__ee_fourier_k50/15000/_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": 1767273352457162637, "commit_timestamp_nsecs": 1767273398350804094, "custom_metadata": {}}

pp__ee_fourier_k50/15000/assets/ur10e/norm_stats.json

@@ -0,0 +1,1525 @@
+{
+  "norm_stats": {
+    "state": {
+      "mean": [
+        0.05313729867339134,
+        0.0037223484832793474,
+        0.106409952044487,
+        0.0008666226058267057,
+        -0.00854077935218811,
+        0.01460699550807476,
+        0.5940247178077698
+      ],
+      "std": [
+        0.1176244392991066,
+        0.16962096095085144,
+        0.07773701101541519,
+        0.04537155479192734,
+        0.06616315245628357,
+        0.6038344502449036,
+        0.49107977747917175
+      ],
+      "q01": [
+        -0.2025573879778385,
+        -0.31106759839057924,
+        0.00890737680345774,
+        -0.11727181684970855,
+        -0.16711108471155167,
+        -1.423702568435669,
+        0.0
+      ],
+      "q99": [
+        0.29901714037060734,
+        0.4115018085956573,
+        0.3217185490995646,
+        0.13774808425903318,
+        0.19241279754638663,
+        1.598234023094177,
+        0.9998
+      ]
+    },
+    "actions": {
+      "mean": [
+        0.0004927651607431471,
+        0.0041174935176968575,
+        0.005378079600632191,
+        -0.0004851108242291957,
+        -0.0010051705176010728,
+        -0.006783927325159311,
+        0.5936462879180908,
+        0.0005593731766566634,
+        -0.00140193453989923,
+        -0.002942675491794944,
+        0.00041861183126457036,
+        0.0004222682327963412,
+        0.0030385160353034735,
+        -0.000329385424265638,
+        -0.0005693831481039524,
+        -0.00029180958517827094,
+        0.000576534541323781,
+        -0.00015302293468266726,
+        -2.4814380594762042e-6,
+        -0.00026009284192696214,
+        0.0003946665092371404,
+        0.000059152920584892854,
+        -0.00006962364568607882,
+        -0.0003498161968309432,
+        0.000053506064432440326,
+        0.00006277022475842386,
+        0.0003592480788938701,
+        -0.00016934213635977358,
+        -0.0001320719748036936,
+        -0.00006360372208291665,
+        0.0001631096820347011,
+        -0.00003750254472834058,
+        -0.000011555765922821593,
+        -0.00006809693149989471,
+        0.00015906375483609736,
+        0.000027621259505394846,
+        -0.00002624911030579824,
+        -0.00012353967758826911,
+        0.000012651072211156134,
+        0.000011669268133118749,
+        0.00012300137314014137,
+        0.00015041981532704085,
+        -0.000056855304137570783,
+        -0.000029746257496299222,
+        0.00007242687570396811,
+        -0.000018986645955010317,
+        -4.556677140499232e-6,
+        -0.00003111416663159616,
+        -0.00009099271846935153,
+        0.000012571626029966865,
+        -0.000013306255823408719,
+        -0.00006410608330043033,
+        6.341869266179856e-6,
+        7.46843352317228e-6,
+        0.00006183652294566855,
+        5.868077096238267e-6,
+        -0.000031927749660098925,
+        -0.00001673644146649167,
+        0.00004087621346116066,
+        -0.000010321637091692537,
+        -2.42824648921669e-6,
+        -0.00001607570084161125,
+        0.00004234428342897445,
+        7.407979410345433e-6,
+        -8.161020559782628e-6,
+        -0.00003870665022986941,
+        3.6879464460071176e-6,
+        4.935619017487625e-6,
+        0.00003818826371571049,
+        0.000051191724196542054,
+        -0.000020595705791492946,
+        -0.000010875675798160955,
+        0.000025950239432859235,
+        -6.903889698151033e-6,
+        -1.2846402341892826e-6,
+        -9.873656381387264e-6,
+        0.00005320658965501934,
+        4.920636911265319e-6,
+        -5.623744073091075e-6,
+        -0.000025824952899711207,
+        2.285133632540237e-6,
+        3.112949343631044e-6,
+        0.00002514909101591911,
+        0.00001734476609271951,
+        -0.000013882496205042116,
+        -7.167188869061647e-6,
+        0.00001790117130440194,
+        -4.3851646296388935e-6,
+        -9.349093375021766e-7,
+        -8.112259820336476e-6,
+        5.336457888915902e-6,
+        3.3939111290237634e-6,
+        -4.019582320324844e-6,
+        -0.00001823278762458358,
+        1.6200752952499897e-6,
+        2.1453445242514135e-6,
+        0.000018565873688203283,
+        0.00004305976108298637,
+        -0.000010276943612552714,
+        -5.3131607273826376e-6,
+        0.000013141003364580683,
+        -3.3397307106497465e-6,
+        -7.699964044149965e-7,
+        -5.316836904967204e-6,
+        -0.00004110661393497139,
+        2.852339093806222e-6,
+        -2.3958800738910213e-6,
+        -0.000013731451872445177,
+        1.5232127452691202e-6,
+        1.386316398566123e-6,
+        0.000013007000234210864,
+        0.00003866312908940017,
+        -7.723581802565604e-6,
+        -3.880812528223032e-6,
+        9.895188668451738e-6,
+        -2.4477972146996763e-6,
+        -5.607461162071559e-7,
+        -3.6636674849432893e-6,
+        0.000010404156455479097,
+        2.0179843431833433e-6,
+        -2.3253292056324426e-6,
+        -0.000010352971003158018,
+        1.0408912203274667e-6,
+        1.1318675205984619e-6,
+        9.978723028325476e-6,
+        -9.606495950720273e-6,
+        -5.949066235189093e-6,
+        -2.8392857984727016e-6,
+        7.4962795224564616e-6,
+        -1.9217382032366004e-6,
+        -4.780694098371896e-7,
+        -3.34773517352005e-6,
+        0.00002107612817781046,
+        1.6454457636427833e-6,
+        -1.817815473259543e-6,
+        -8.123145562421996e-6,
+        7.675727147216094e-7,
+        9.284493671657401e-7,
+        7.868961802159902e-6,
+        -9.203896524923039e-7,
+        -4.736490154755302e-6,
+        -2.4710741399758263e-6,
+        6.142801339592552e-6,
+        -1.6350784335372737e-6,
+        -4.335802543664613e-7,
+        -1.4092432820689282e-6,
+        0.00001656184576859232,
+        1.3327525039130705e-6,
+        -1.3708133792533772e-6,
+        -6.531353847094579e-6,
+        5.90599142924475e-7,
+        6.878227623019484e-7,
+        6.5318749875586946e-6,
+        -4.229010301060043e-6,
+        -4.000812168669654e-6,
+        -2.2653048290521838e-6,
+        5.042587417847244e-6,
+        -1.4821022205069312e-6,
+        -3.451863506143127e-7,
+        -2.389324890827993e-6,
+        8.16222745925188e-6,
+        1.1428018069636892e-6,
+        -1.1332826943544205e-6,
+        -5.504417913471116e-6,
+        5.024350002713618e-7,
+        5.467733217301429e-7,
+        5.279353445075685e-6,
+        -0.0000211214974115137,
+        -3.1678093819209607e-6,
+        -1.8488366322344518e-6,
+        4.2470492189750075e-6,
+        -9.783543646335602e-7,
+        -1.83213785476255e-7,
+        -1.4789864053454949e-6,
+        0.000013087515981169418,
+        7.242887818392774e-7,
+        -1.0046998113466543e-6,
+        -4.340507530287141e-6,
+        3.3989198300332646e-7,
+        5.470611199598352e-7,
+        4.507525773078669e-6,
+        -0.000016270019841613248,
+        -2.829020104400115e-6,
+        -1.296591335631092e-6,
+        3.456672175161657e-6,
+        -9.017520028464787e-7,
+        -1.4440175277741218e-7,
+        -1.4689791214550496e-6,
+        0.000021527563148993067,
+        5.846809472132009e-7,
+        -5.933143256697804e-7,
+        -3.612318550949567e-6,
+        3.5949764765064174e-7,
+        4.328513796281186e-7,
+        3.6886276575387456e-6,
+        -0.000021549723896896467,
+        -2.1144403490325203e-6,
+        -1.4159754755382892e-6,
+        2.68504163614125e-6,
+        -7.679741429456044e-7,
+        -5.776412592695124e-8,
+        -1.192243757941469e-6,
+        0.00003148229370708577,
+        5.478635785038932e-7,
+        -6.972362598389736e-7,
+        -3.211202283637249e-6,
+        2.2451455095051642e-7,
+        3.183014598562295e-7,
+        2.939946398328175e-6,
+        4.565895324049052e-6,
+        -1.7405694734407007e-6,
+        -9.741680742081371e-7,
+        2.2756712496629916e-6,
+        -5.214625389271532e-7,
+        -2.7989250384052866e-7,
+        -8.445824164482474e-7,
+        0.00003105050200247206,
+        5.799903988190636e-7,
+        -7.536694397458632e-7,
+        -2.5344147616124246e-6,
+        2.630972346651106e-7,
+        2.9163163617340615e-7,
+        2.3315690214076312e-6,
+        0.000028919430405949242,
+        -1.4791980902373325e-6,
+        -7.087014637363609e-7,
+        2.1661373921233462e-6,
+        -5.724799052586604e-7,
+        -3.0745661661057966e-7,
+        -5.235916091805848e-7,
+        0.000011437900866440032,
+        3.404710184895521e-7,
+        -4.5958552163938293e-7,
+        -2.1246926280582557e-6,
+        2.5418461291337735e-7,
+        3.871034266467177e-7,
+        1.9572280507418327e-6,
+        0.00003196126999682747,
+        -1.3085544878777e-6,
+        -6.04459614805819e-7,
+        1.575823830535228e-6,
+        -4.1182735799338843e-7,
+        -1.1744142369707333e-7,
+        -1.3322799077286618e-6,
+        2.7163066533830715e-6,
+        3.6782887491426663e-7,
+        -2.6708795530794305e-7,
+        -1.72060231307114e-6,
+        1.8409495794458053e-7,
+        3.7163931665418204e-7,
+        1.8107581354342983e-6,
+        0.000013270215276861563,
+        -1.1510371678014053e-6,
+        -6.630299367316184e-7,
+        1.070535518010729e-6,
+        -3.826233978543314e-7,
+        1.1823525625231923e-7,
+        -1.049707634592778e-6,
+        -2.8137444587628124e-6,
+        2.6974137767865614e-7,
+        -4.0497872078049113e-7,
+        -1.4479496712738182e-6,
+        3.8844518712721765e-8,
+        2.0770157149740953e-8,
+        1.7100152263083146e-6,
+        -0.00002849056909326464,
+        -9.183266342915886e-7,
+        -3.5311214219291287e-7,
+        1.1368990726623451e-6,
+        -3.410139868265105e-7,
+        -2.7979771743957826e-7,
+        -9.036052119881788e-7,
+        -8.269290447060484e-6,
+        3.0624855185124034e-7,
+        -1.1787745535229988e-7,
+        -1.2770473176715313e-6,
+        1.6275393477371836e-7,
+        3.3039643199117563e-7,
+        1.3103665423841449e-6,
+        -0.00001745441295497585,
+        -6.018551061970356e-7,
+        -2.794714930587361e-7,
+        9.792224773264024e-7,
+        -2.040594893060188e-7,
+        -3.4366582468692286e-8,
+        -4.681474479184544e-7,
+        -3.3097076084231958e-6,
+        1.669202376319845e-8,
+        -2.7714148131963157e-7,
+        -9.64512764767278e-7,
+        1.1432685909085194e-7,
+        1.4834664341378812e-7,
+        1.1487189794934238e-6,
+        -0.000010153660696232691,
+        -6.140094228612725e-7,
+        -5.253555173112545e-7,
+        5.710655841539847e-7,
+        -1.404835359153367e-7,
+        4.7323091223461233e-8,
+        -5.46001842849364e-7,
+        -1.2161887070760713e-6,
+        1.2690516371094418e-7,
+        -1.3310798863130913e-7,
+        -7.572793379040377e-7,
+        1.7468086355165724e-7,
+        3.3051062331423964e-8,
+        7.131747565836122e-7,
+        -0.00003459135041339323,
+        -4.08122900807939e-7,
+        -1.8256875478073198e-7,
+        5.756255063715798e-7,
+        -1.6261435575870564e-7,
+        -2.021908862559485e-8,
+        -4.1869603251143417e-7,
+        -1.589809244251228e-6,
+        1.1844038283470582e-7,
+        -7.242162070042468e-8,
+        -4.0011059354583267e-7,
+        -2.4760929306921753e-8,
+        -1.3480507732310798e-7,
+        1.0380339290350094e-6,
+        1.6824413933136384e-6,
+        -4.5796767267347605e-7,
+        -2.0989804738746898e-7,
+        4.763822971654008e-7,
+        -1.7666546625605406e-7,
+        -2.998827852707109e-8,
+        3.5782920804194873e-7,
+        -0.00005497613165061921,
+        -4.8836572119626e-8,
+        -2.8484058134381485e-7,
+        -2.0204961970193835e-7,
+        -1.9280198770843526e-8,
+        8.62186144701127e-9,
+        3.8213460129554733e-7,
+        1.619790509721497e-6,
+        -1.449097339900618e-7,
+        -1.7092831683385157e-7,
+        3.423832595217391e-7,
+        -9.912067611139719e-8,
+        5.862289853553193e-8,
+        -7.740770513464668e-8,
+        -7.812944204488304e-6,
+        1.33551338876714e-7,
+        1.7766551252407226e-7,
+        -2.63828656699161e-8,
+        -5.126612023786947e-9,
+        -4.134697562108158e-8,
+        8.136574791706153e-8,
+        -4.875783361057984e-6
+      ],
+      "std": [
+        0.054471205919981,
+        0.07741593569517136,
+        0.06635452806949615,
+        0.02219589799642563,
+        0.03713744506239891,
+        0.2155904769897461,
+        0.44884487986564636,
+        0.022129571065306664,
+        0.030908629298210144,
+        0.027230359613895416,
+        0.009623671881854534,
+        0.015185607597231865,
+        0.08782459795475006,
+        0.11000800132751465,
+        0.0054918802343308926,
+        0.007706818170845509,
+        0.007092009298503399,
+        0.004339421167969704,
+        0.006056886166334152,
+        0.025186194106936455,
+        0.0549297109246254,
+        0.0023341968189924955,
+        0.003104041563346982,
+        0.0028293700888752937,
+        0.002068394096568227,
+        0.0026343404315412045,
+        0.01149396225810051,
+        0.03640395402908325,
+        0.0011078902753069997,
+        0.0015103736659511924,
+        0.0014142218278720975,
+        0.0010499481577426195,
+        0.0013543356908485293,
+        0.005356063600629568,
+        0.027538813650608063,
+        0.0007573548937216401,
+        0.001025661244057119,
+        0.0009198821499012411,
+        0.0005544671439565718,
+        0.0007547291461378336,
+        0.0035150113981217146,
+        0.02213539369404316,
+        0.000455506524303928,
+        0.0006300596869550645,
+        0.0005910721374675632,
+        0.00034599038190208375,
+        0.0004919870407320559,
+        0.002043767599388957,
+        0.018490547314286232,
+        0.00037283002166077495,
+        0.0005109719350002706,
+        0.00045568859786726534,
+        0.00023278511071112007,
+        0.00032812528661452234,
+        0.0016570856096222997,
+        0.015843955799937248,
+        0.0002481868432369083,
+        0.0003480722662061453,
+        0.000325335975503549,
+        0.00017280328029301018,
+        0.0002512190258130431,
+        0.0010802638716995716,
+        0.01408928632736206,
+        0.00022223185806069523,
+        0.00030612919363193214,
+        0.0002718937466852367,
+        0.00012990496179554611,
+        0.00018422518041916192,
+        0.000973312184214592,
+        0.012508493848145008,
+        0.00015761872055009007,
+        0.00022044251090846956,
+        0.0002061627892544493,
+        0.0001057182380463928,
+        0.00015497820277232677,
+        0.0006764288409613073,
+        0.011296260170638561,
+        0.00014791855937801301,
+        0.00020399357890710235,
+        0.00018064411415252835,
+        0.00008491610060445964,
+        0.0001210872214869596,
+        0.0006432573427446187,
+        0.01023485790938139,
+        0.00010906582610914484,
+        0.00015326884749811143,
+        0.000142823948408477,
+        0.00007245635788422078,
+        0.00010654709330992773,
+        0.00046615453902632,
+        0.009492816403508186,
+        0.00010573185136308894,
+        0.0001460392086300999,
+        0.0001292345259571448,
+        0.00006049493094906211,
+        0.00008650264499010518,
+        0.00045887212036177516,
+        0.008779959753155708,
+        0.00007973915489856154,
+        0.00011203208850929514,
+        0.00010434568685013801,
+        0.00005302290082909167,
+        0.00007767079659970477,
+        0.0003416067629586905,
+        0.00814682524651289,
+        0.00007915469905128703,
+        0.00010886993550229818,
+        0.0000966577063081786,
+        0.000045318782213144004,
+        0.00006455113179981709,
+        0.0003425226896069944,
+        0.007630111649632454,
+        0.0000611644281889312,
+        0.00008592260564910248,
+        0.00007935166649986058,
+        0.000040576029277872294,
+        0.00005935502122156322,
+        0.00026016749325208366,
+        0.007179272826761007,
+        0.00006150962872197852,
+        0.00008518873801222071,
+        0.00007461248605977744,
+        0.0000355022020812612,
+        0.0000502788316225633,
+        0.00026474526384845376,
+        0.006768743507564068,
+        0.000048591213271720335,
+        0.00006827081961091608,
+        0.00006262768874876201,
+        0.000032297353754984215,
+        0.000046926012146286666,
+        0.00020619359565898776,
+        0.006469320971518755,
+        0.00004949716094415635,
+        0.00006947917427169159,
+        0.00005963100556982681,
+        0.000028536098398035392,
+        0.000040751590859144926,
+        0.00021295368787832558,
+        0.006215229630470276,
+        0.00003966964141000062,
+        0.00005751234493800439,
+        0.00005083374344394542,
+        0.00002690069413802121,
+        0.00003854740134556778,
+        0.00016899738693609834,
+        0.005915363319218159,
+        0.00004071096918778494,
+        0.00005744508234784007,
+        0.000049411490181228146,
+        0.000024393977582803927,
+        0.000034359978599241003,
+        0.00017412411398254335,
+        0.005679433699697256,
+        0.000034098236938007176,
+        0.00004947670822730288,
+        0.0000428831554017961,
+        0.000023318862076848745,
+        0.0000333583120664116,
+        0.00014037704386282712,
+        0.005502832122147083,
+        0.000035476441553328186,
+        0.00005029448220739141,
+        0.000041998053347924724,
+        0.000021735353584517725,
+        0.00003051272506127134,
+        0.00014704464410897344,
+        0.005227266810834408,
+        0.000031422849133377895,
+        0.0000454086039098911,
+        0.000037437119317473844,
+        0.000021104986444697715,
+        0.0000290283169306349,
+        0.00012362531560938805,
+        0.005091419909149408,
+        0.00003195118915755302,
+        0.00004717931369668804,
+        0.00003730821481440216,
+        0.000018998216546606272,
+        0.00002651557770150248,
+        0.0001294043322559446,
+        0.004971646703779697,
+        0.00002650190617714543,
+        0.000040163333324017,
+        0.00003248209395678714,
+        0.000018027179976343177,
+        0.00002582307206466794,
+        0.00010801247117342427,
+        0.004754771012812853,
+        0.00002661882717802655,
+        0.00003864263999275863,
+        0.00003127731179120019,
+        0.000016530248103663325,
+        0.000023721948309685104,
+        0.00010995724005624652,
+        0.004578577820211649,
+        0.000022862250261823647,
+        0.000033582466130610555,
+        0.000027823365599033423,
+        0.000015542551409453154,
+        0.000022585352780879475,
+        0.00009167923417408019,
+        0.004508121404796839,
+        0.00002280540320498403,
+        0.000033627373341005296,
+        0.00002707460953388363,
+        0.000014449000445893034,
+        0.00002091616806865204,
+        0.0000951679321588017,
+        0.004384348168969154,
+        0.00002045510154857766,
+        0.00002992084773723036,
+        0.000024755014237598516,
+        0.000014510045730276033,
+        0.000020438790670596063,
+        0.00008195763803087175,
+        0.004295016173273325,
+        0.000021367646695580333,
+        0.00003022674354724586,
+        0.000024624087018310092,
+        0.000013480367670126725,
+        0.000019035556761082262,
+        0.0000849158750497736,
+        0.004229281097650528,
+        0.000018941680536954664,
+        0.000027560265152715147,
+        0.000022634478227701038,
+        0.000013048511391389184,
+        0.00001906735087686684,
+        0.00007548843132099137,
+        0.004119527991861105,
+        0.00002021879299718421,
+        0.00002808637873386033,
+        0.000022752908989787102,
+        0.000012890232028439641,
+        0.000019045859517063946,
+        0.0000775336375227198,
+        0.004028849303722382,
+        0.00001860819975263439,
+        0.000026566647647996433,
+        0.000021319850930012763,
+        0.000012895960026071407,
+        0.00001921564035001211,
+        0.0000719531744834967,
+        0.003980362322181463,
+        0.000019542083464330062,
+        0.000027795384085038677,
+        0.000021858841137145646,
+        0.000013172734725230839,
+        0.000019754204913624562,
+        0.00007459476182702929,
+        0.003902778960764408,
+        0.00002043036147370003,
+        0.00003195027602487244,
+        0.000024560755264246836,
+        0.000014010387531016022,
+        0.00002065290027530864,
+        0.00007676216046093032,
+        0.0038293309044092894,
+        0.000023211092411656864,
+        0.00003743018169188872,
+        0.000025842035029199906,
+        0.000015010549759608693,
+        0.000020894838598906063,
+        0.00008554229862056673,
+        0.0037848083302378654,
+        0.00002203849180659745,
+        0.00003869964712066576,
+        0.00002533109909563791,
+        0.000013811703865940217,
+        0.000020183759261271916,
+        0.0000796892709331587,
+        0.0037312512286007404,
+        0.000020139470507274382,
+        0.00003185970854246989,
+        0.00002326746835024096,
+        0.000014411976735573262,
+        0.000020371249775053002,
+        0.00007360208837781101,
+        0.0036642765626311302,
+        0.000017049775124178268,
+        0.000027688640329870395,
+        0.00001989392148971092,
+        0.000012698577847913839,
+        0.000017972832210944034,
+        0.00006560993642779067,
+        0.0036550285294651985,
+        0.000016028958270908333,
+        0.000024884970116545446,
+        0.00001818954842747189,
+        0.000011278674719505943,
+        0.000015812189303687774,
+        0.00006195339665282518,
+        0.0036316923797130585,
+        0.00001560178498039022,
+        0.000024082159143290482,
+        0.00001761745625117328,
+        0.00001089257057174109,
+        0.000015621577404090203,
+        0.00005938451431575231,
+        0.003624820150434971,
+        0.000014879788068355992,
+        0.00002372561357333325,
+        0.00001737473940011114,
+        0.00001018085913528921,
+        0.000015368110325653106,
+        0.00005791655712528154,
+        0.0035625966265797615,
+        0.000014492210539174266,
+        0.000023905164198367856,
+        0.000016788300854386762,
+        0.00001016644364426611,
+        0.000014378600099007599,
+        0.00005657917427015491,
+        0.0035458113998174667,
+        0.000014646487215941306,
+        0.000023379501726594754,
+        0.000016420719475718215,
+        0.000010098652637680061,
+        0.000014440963241213467,
+        0.00005597082417807542,
+        0.0035429371055215597,
+        0.000014022874893271364,
+        0.00002083050821966026,
+        0.000015985659047146328,
+        0.00001016553051158553,
+        0.000013916131138103083,
+        0.000056019474868662655,
+        0.0034998103510588408,
+        0.000014245401871448848,
+        0.000022099453417467885,
+        0.00001657450047787279,
+        0.000010621713045111392,
+        0.000014651923265773803,
+        0.00005747186514781788,
+        0.0034743566066026688,
+        0.000014478311641141772,
+        0.00002349081842112355,
+        0.000017406200640834868,
+        0.000010810344974743202,
+        0.000014864897821098566,
+        0.00005688491728506051,
+        0.0034770709462463856,
+        0.000016515868992428295,
+        0.00002587365270301234,
+        0.000018931375961983576,
+        0.000010814174856932368,
+        0.000016132373275468126,
+        0.00006046127964509651,
+        0.003478288883343339
+      ],
+      "q01": [
+        -0.15909142580032348,
+        -0.23591104531884194,
+        -0.1173719973474741,
+        -0.0725411132991314,
+        -0.12759929532110692,
+        -0.6792203711509704,
+        0.0,
+        -0.06588815684318543,
+        -0.1043403072744608,
+        -0.07569241938591004,
+        -0.026123490297794344,
+        -0.04355562012195587,
+        -0.26860033273696904,
+        -0.31594268302917483,
+        -0.017998227071762084,
+        -0.025448361220955847,
+        -0.021794256627559662,
+        -0.014516740383952856,
+        -0.017180888187885286,
+        -0.08126688668131828,
+        -0.15649057692885399,
+        -0.006211889632977545,
+        -0.009136751361563802,
+        -0.00862356810811907,
+        -0.00636542599722743,
+        -0.007391496415808798,
+        -0.035477512252330784,
+        -0.10543232785761356,
+        -0.003451917093619704,
+        -0.004779039949178695,
+        -0.00400189593732357,
+        -0.003541165339201688,
+        -0.003899161383137107,
+        -0.016611159443855286,
+        -0.07849292940199376,
+        -0.001959954298567027,
+        -0.0028813693393021818,
+        -0.0027899963006842885,
+        -0.0015735435530543325,
+        -0.002058688412979245,
+        -0.010417635813355447,
+        -0.0639662309139967,
+        -0.0013655057244002819,
+        -0.0019750579357147215,
+        -0.0016051418443210422,
+        -0.0011298993272706869,
+        -0.0014582404736429451,
+        -0.006558569058775902,
+        -0.05251251416504383,
+        -0.0009643727889051661,
+        -0.0014315907768905163,
+        -0.001390606227517128,
+        -0.0006525630556046963,
+        -0.0008888287223875523,
+        -0.004895182948186994,
+        -0.0455008171916008,
+        -0.0007486765272915363,
+        -0.0010981693975627422,
+        -0.000882065705396235,
+        -0.0005725251567782834,
+        -0.0007578355811536312,
+        -0.0034509299120865766,
+        -0.04015725544691086,
+        -0.0005749710650416091,
+        -0.0008555282396264373,
+        -0.0008345671106129884,
+        -0.0003548953181132674,
+        -0.0004969418092630804,
+        -0.0028254041438922285,
+        -0.035569155138731004,
+        -0.0004705392210977152,
+        -0.0006840464788489043,
+        -0.0005612756710499526,
+        -0.00035702317333780227,
+        -0.0004648537166416645,
+        -0.002128881996870041,
+        -0.03083838468044997,
+        -0.00038048800966935235,
+        -0.000566111233481206,
+        -0.0005538229705183766,
+        -0.00023378437627106905,
+        -0.00032636490603908896,
+        -0.0018706658044829966,
+        -0.02931403119564057,
+        -0.00032479510342236606,
+        -0.000475912622641772,
+        -0.00038455819835653526,
+        -0.00024285348085686562,
+        -0.0003135723091545515,
+        -0.001452100250311196,
+        -0.026792802512645722,
+        -0.00027214939426630733,
+        -0.00040756029058247796,
+        -0.0003960524151916616,
+        -0.00016536210232879968,
+        -0.00023339646961539982,
+        -0.0013170160830486568,
+        -0.02499642580747604,
+        -0.00024108570310054343,
+        -0.000348888192162849,
+        -0.0002832006874959916,
+        -0.00017626483961939813,
+        -0.00023037131156306717,
+        -0.0010707858602516352,
+        -0.023444418999552727,
+        -0.00020677300081588327,
+        -0.0003006568572251126,
+        -0.0002946901318966411,
+        -0.00012258605094393713,
+        -0.00017606193688698117,
+        -0.0009890523520298302,
+        -0.02203794540464878,
+        -0.0001841578755469527,
+        -0.00026803012369200585,
+        -0.00021574946893379094,
+        -0.00013577964630676436,
+        -0.0001734850220615044,
+        -0.0008068450814578683,
+        -0.020441363253444433,
+        -0.00016045538827893325,
+        -0.000241462552268058,
+        -0.00022808200246654452,
+        -0.00009496491550235078,
+        -0.00013633702206425368,
+        -0.0007666170152835548,
+        -0.019491385824978353,
+        -0.0001477667265571654,
+        -0.00021141915648477145,
+        -0.0001735861329594627,
+        -0.00010635797707363962,
+        -0.00013736910806037486,
+        -0.0006427704328671098,
+        -0.018371846283972263,
+        -0.0001293026403174736,
+        -0.00019473347970051693,
+        -0.0001822685490187723,
+        -0.00007863951541366989,
+        -0.00010972291373182087,
+        -0.0006055113250389692,
+        -0.017673968444019555,
+        -0.00011848664400167762,
+        -0.00017848168123746295,
+        -0.00014009435730986296,
+        -0.0000859230127884075,
+        -0.00011260414165444673,
+        -0.0005137789773289115,
+        -0.016190811094641686,
+        -0.00010690572904422879,
+        -0.0001638943360070698,
+        -0.0001501999743981287,
+        -0.00006396331703872426,
+        -0.00009358972666086629,
+        -0.0005007586104329676,
+        -0.01619275164604187,
+        -0.00010315810496686024,
+        -0.00015018478848505767,
+        -0.00011576190601917915,
+        -0.00007506265956908466,
+        -0.00009504533028230073,
+        -0.0004348065205849708,
+        -0.01565815359354019,
+        -0.00009185501914471398,
+        -0.0001392430538544431,
+        -0.00012555517936125394,
+        -0.00005638334401883189,
+        -0.00008002441792050375,
+        -0.000424562167050317,
+        -0.015019814673811197,
+        -0.00009095547425094991,
+        -0.00013837829248514026,
+        -0.00010207304285722785,
+        -0.00006334332783007992,
+        -0.00008174893613904723,
+        -0.0003743791728978976,
+        -0.014377124969661235,
+        -0.00008569652531296014,
+        -0.00013066508993506436,
+        -0.00010974786025471983,
+        -0.000049689067096915106,
+        -0.00007135935623664408,
+        -0.0003742531020194291,
+        -0.014187388128042218,
+        -0.0000794089021859691,
+        -0.00011592403496615584,
+        -0.0000893614823697135,
+        -0.0000560631047235802,
+        -0.0000719853327609599,
+        -0.0003315736224874853,
+        -0.013546264378726483,
+        -0.0000717077999375761,
+        -0.00010777361229993398,
+        -0.0000922710225568153,
+        -0.00004497638295288198,
+        -0.00006191108026541768,
+        -0.0003081054835114628,
+        -0.013243352345749734,
+        -0.00006864526765421036,
+        -0.0001027144630090333,
+        -0.00008054810634930621,
+        -0.000049568866193294524,
+        -0.00006090367061551658,
+        -0.00028183929068036377,
+        -0.012778499507904054,
+        -0.00006117501191329209,
+        -0.00009796037861378861,
+        -0.00008128996908781118,
+        -0.00004006202814634888,
+        -0.00005725241685286168,
+        -0.0002700211696093902,
+        -0.012565325374901296,
+        -0.00006034678600262851,
+        -0.0000907877254066989,
+        -0.00007010409524082204,
+        -0.000041811013757251194,
+        -0.0000556492382194847,
+        -0.0002493678867351264,
+        -0.011774876400828364,
+        -0.00005579026520717887,
+        -0.00008717787461355332,
+        -0.00007443824684014546,
+        -0.0000356287220492959,
+        -0.00005372272633831019,
+        -0.00024090840241406124,
+        -0.011997990143299102,
+        -0.000058186027396004656,
+        -0.00008279621332185345,
+        -0.00006459413380362097,
+        -0.00003831381897907702,
+        -0.00005259611281217076,
+        -0.00022310287454165522,
+        -0.011667696542292834,
+        -0.00005284995273686948,
+        -0.00008384203007444737,
+        -0.00006744749160716312,
+        -0.000032252046390203774,
+        -0.0000481185418262612,
+        -0.00021629847201984375,
+        -0.011543062198162084,
+        -0.00005847320245811716,
+        -0.00008175599463284015,
+        -0.00006548594534397124,
+        -0.000038102848210837704,
+        -0.0000492544468608685,
+        -0.00021983185317367318,
+        -0.011402166564017536,
+        -0.00005387163197156041,
+        -0.0000825874566333369,
+        -0.00006687336476170457,
+        -0.000033979644608916735,
+        -0.00005101760845864191,
+        -0.00020870170197449626,
+        -0.011224610126763582,
+        -0.00006089058108045721,
+        -0.00008930324914399536,
+        -0.00007119371397420766,
+        -0.00003768286973936482,
+        -0.000053939796844497314,
+        -0.0002299774368293584,
+        -0.010891138190776108,
+        -0.000060360829543788016,
+        -0.00009397184122353786,
+        -0.00007481435872614382,
+        -0.00003693910902366043,
+        -0.00005306754379416818,
+        -0.0002138554411940277,
+        -0.01082267729640007,
+        -0.00005871810513781385,
+        -0.00008991035101935281,
+        -0.00007033640673034824,
+        -0.000036662435624748536,
+        -0.00005430486062541601,
+        -0.0002200629871338608,
+        -0.010579705061763527,
+        -0.000053560086898505666,
+        -0.00008582322522997861,
+        -0.00007200759416446087,
+        -0.000034787034674081956,
+        -0.000048553584120236346,
+        -0.0002130190141499044,
+        -0.010566252829506993,
+        -0.00005062428400851786,
+        -0.00007914942941861225,
+        -0.000059331482945708576,
+        -0.00003430225813062863,
+        -0.00004731270743068308,
+        -0.00019290218278765676,
+        -0.010022874112427235,
+        -0.000047412463498767456,
+        -0.00007506768405437467,
+        -0.00005783305459190161,
+        -0.00002997652450576426,
+        -0.00004301143199554646,
+        -0.00018155146474018715,
+        -0.010345511507242918,
+        -0.00004604501994326713,
+        -0.00007326857978478065,
+        -0.00005605542755220084,
+        -0.000028739307308569553,
+        -0.00004117103526368737,
+        -0.0001768201191909612,
+        -0.010309934300184249,
+        -0.00004275882025249304,
+        -0.00006731156632304187,
+        -0.00005409912285394967,
+        -0.000026329884584993103,
+        -0.00004337953124195339,
+        -0.00016598801976069823,
+        -0.01018075596317649,
+        -0.000043296541273593885,
+        -0.00006625080597586939,
+        -0.00005305007540155199,
+        -0.000028063857986126094,
+        -0.000040029496920760744,
+        -0.00017517980523407465,
+        -0.010162209048867225,
+        -0.0000426195073465351,
+        -0.00006396621902240442,
+        -0.000051819634152343525,
+        -0.000029033373587299122,
+        -0.00004320319109829139,
+        -0.00016554977782070636,
+        -0.010125353442132473,
+        -0.000045235728612169645,
+        -0.00006515037268400192,
+        -0.00004881905938964337,
+        -0.000028501104679889977,
+        -0.00004041612264700232,
+        -0.00016903663412667815,
+        -0.010062240642309189,
+        -0.000043975374847650503,
+        -0.00006714530037716033,
+        -0.00005093284538015727,
+        -0.00002801080799545163,
+        -0.00004157804111018775,
+        -0.00016746679800562535,
+        -0.009975232541561126,
+        -0.000045126923860516424,
+        -0.00007225614520721142,
+        -0.00005288218818604951,
+        -0.000029170855134725583,
+        -0.000040984369814395916,
+        -0.0001663898449391129,
+        -0.009851353891193867,
+        -0.00004605035483837126,
+        -0.00007112415134906766,
+        -0.0000582737371325493,
+        -0.000028596166521310774,
+        -0.000044461411982774806,
+        -0.0001793744623661041,
+        -0.009932486386597156
+      ],
+      "q99": [
+        0.16212226042747496,
+        0.25229084351658815,
+        0.18486946655809877,
+        0.061344154804944984,
+        0.10886132830679413,
+        0.6657196632862092,
+        0.9998,
+        0.06576045138835906,
+        0.09603128664791583,
+        0.045872775423526774,
+        0.03236173896193503,
+        0.05487722767591477,
+        0.2764705369472503,
+        0.3158153381347656,
+        0.014891946853697303,
+        0.023800758841633796,
+        0.018585946798324586,
+        0.012693745650351043,
+        0.021670551879704,
+        0.08171248322725297,
+        0.15637542012631894,
+        0.0069066042726859435,
+        0.009335029612295327,
+        0.0047039436949416966,
+        0.006996506480127576,
+        0.008702572771534321,
+        0.036967494320869446,
+        0.1053792189329863,
+        0.002857540221139789,
+        0.004296822899580001,
+        0.003630450583249331,
+        0.003087930056080223,
+        0.004339818637073041,
+        0.01760330287218094,
+        0.07830542102754112,
+        0.0021181412500329315,
+        0.0029414672447368503,
+        0.0015112007031682874,
+        0.0019269651249051098,
+        0.0024494722973555337,
+        0.01144827667623758,
+        0.06380807581543924,
+        0.0011728783883154393,
+        0.001813931218627841,
+        0.0014501392262987797,
+        0.0009983487627469,
+        0.0015205995496362446,
+        0.006439514055848122,
+        0.05315425983667374,
+        0.0010328240715665742,
+        0.0014663059995509691,
+        0.0007404281431809068,
+        0.0008001560054719447,
+        0.001095746040716767,
+        0.0052222565246745935,
+        0.04538371631056071,
+        0.0006383903683163227,
+        0.0009848102391697465,
+        0.000784420038573444,
+        0.0004887675133766608,
+        0.0007893593505024911,
+        0.0033699059166945513,
+        0.039476715770363824,
+        0.0006163579971762373,
+        0.0008800703473389147,
+        0.00044086696542799475,
+        0.000447366009419784,
+        0.0006247742644743994,
+        0.0030928143300116076,
+        0.035530846929550164,
+        0.0004072732983855529,
+        0.0006279987238347533,
+        0.0004935565162450075,
+        0.0003003585597965866,
+        0.0004761313281953334,
+        0.002095991748571396,
+        0.030764527727663513,
+        0.00040901676943758494,
+        0.0005783375784289091,
+        0.00029091126989806075,
+        0.0002924530396237969,
+        0.0004053453548112883,
+        0.0020267377858981485,
+        0.02924695993065833,
+        0.0002835550721851177,
+        0.0004330007825046778,
+        0.0003438553759129719,
+        0.00021280083870515228,
+        0.0003332204633043148,
+        0.0014287737663835282,
+        0.02666576117277146,
+        0.00029395890273153794,
+        0.00041647118404507634,
+        0.0002144097874115686,
+        0.00020652228510007256,
+        0.0002845425325096586,
+        0.0014505576011724767,
+        0.024956510066986085,
+        0.00020434754303423695,
+        0.00031225532339885847,
+        0.0002542992958333343,
+        0.0001512215134222061,
+        0.00024444468863075595,
+        0.0010491569862701002,
+        0.023069663181155914,
+        0.00022068456518463788,
+        0.00031388382902368915,
+        0.00016011388470651587,
+        0.00015496468472993004,
+        0.00021093179946765303,
+        0.001086882085958496,
+        0.0219863745495677,
+        0.00016183016332215621,
+        0.00024126361515372995,
+        0.0001919538950081915,
+        0.00011462044673971831,
+        0.00018409015340730545,
+        0.0008034803080838172,
+        0.020384925924241543,
+        0.0001725941591721494,
+        0.00024417606623610485,
+        0.00012369809069205076,
+        0.00012043028190964833,
+        0.00016321412267861885,
+        0.0008335524674970658,
+        0.019483701366186136,
+        0.00012494574459269637,
+        0.0001922543402411975,
+        0.00015176694963593034,
+        0.00008988425261923113,
+        0.00014705147650092838,
+        0.0006307619461789725,
+        0.01831897035390139,
+        0.00013962995521724217,
+        0.00019706185359973463,
+        0.00010126548151019962,
+        0.0000950566966785118,
+        0.00013232208000263205,
+        0.000660218992456794,
+        0.01764387315139175,
+        0.00010247177979908883,
+        0.00015943087645573536,
+        0.00012344940464245158,
+        0.00007527396054356363,
+        0.00011956113649066536,
+        0.0005160252890083939,
+        0.016168088349699967,
+        0.00011438966927817092,
+        0.00016185711201978847,
+        0.00008490921122720466,
+        0.00007747934755752788,
+        0.00010678794741397725,
+        0.0005481199477799237,
+        0.01615487493276596,
+        0.00008468688050634237,
+        0.00013357372635509814,
+        0.00010555801286245698,
+        0.00006330196186900142,
+        0.000101998839713633,
+        0.00042467831117101035,
+        0.015395162180066112,
+        0.00009998356355354183,
+        0.00014026010758243506,
+        0.00007274458486353978,
+        0.00006777426579501484,
+        0.00009303196137771012,
+        0.0004640237890882414,
+        0.01499328883215785,
+        0.00007716326334048071,
+        0.00011421011368511244,
+        0.00009314750883495434,
+        0.00005738810400944202,
+        0.00008783590141683817,
+        0.0003717806453816595,
+        0.014554040935635573,
+        0.00008682058225385845,
+        0.00012317761406302453,
+        0.00007470068163238468,
+        0.00005833794826176018,
+        0.0000807305663009174,
+        0.0003966953692026436,
+        0.014136480166763069,
+        0.00006818514921469611,
+        0.00010206100093200813,
+        0.00008256861115805807,
+        0.000049349277792498495,
+        0.00007737125915009528,
+        0.00031947106840088976,
+        0.013474849896132943,
+        0.00007525306886527682,
+        0.00010829368010163308,
+        0.00006185477034305223,
+        0.000050571078446228043,
+        0.00007023279787972571,
+        0.0003371094720438124,
+        0.013198494754731657,
+        0.000059138852264732124,
+        0.00008794275637483227,
+        0.00007062362715369087,
+        0.00004125927500426771,
+        0.00006884480739245192,
+        0.0002681864194571972,
+        0.012739881089329719,
+        0.00006564935145433995,
+        0.00009313276901375502,
+        0.00005367935128742829,
+        0.00004326853083912283,
+        0.000059486858663149135,
+        0.000290332669764757,
+        0.012545088732242582,
+        0.00005332418816396964,
+        0.0000798318615881726,
+        0.00006209178492426877,
+        0.00003812435691943395,
+        0.00005920125464908778,
+        0.00024440587060526003,
+        0.011752966091036793,
+        0.00006044615641003474,
+        0.00008350273473188281,
+        0.00005087533958721905,
+        0.00004026508983224629,
+        0.00005588843687437471,
+        0.0002512834585038945,
+        0.01198878407478332,
+        0.00004965451551834123,
+        0.00007573206002125512,
+        0.00006231120838783678,
+        0.00003364615770988169,
+        0.00005282912561669945,
+        0.0002209204529877752,
+        0.011624232469499118,
+        0.00005783643363974989,
+        0.00008142779953777802,
+        0.00005095888577052386,
+        0.000036499108670977874,
+        0.000052707964507862936,
+        0.000231120909564197,
+        0.01148185554742813,
+        0.00004983878775965424,
+        0.00007604824067093425,
+        0.000057698829320725065,
+        0.000033903982862830136,
+        0.000054097583203110825,
+        0.00020294103957712662,
+        0.01119744649901986,
+        0.00005782577866921202,
+        0.00007854039029916743,
+        0.0000542719262011815,
+        0.00003704023372847582,
+        0.00005206325108883899,
+        0.00022946770491544147,
+        0.011211802294850345,
+        0.0000541387221601326,
+        0.00008076134007424106,
+        0.00006221706023206938,
+        0.00003468251589220018,
+        0.000056180346361361434,
+        0.00020245690247975277,
+        0.010855805838853122,
+        0.00005655899195699014,
+        0.0000880048372084274,
+        0.00006508456927258521,
+        0.00003773696059361102,
+        0.00005296763135702342,
+        0.00022277740947902205,
+        0.010783490767329935,
+        0.00005639911381294957,
+        0.00008836033511906856,
+        0.00007074359970283693,
+        0.00003449545800685873,
+        0.00004960680939257152,
+        0.0002082344447262587,
+        0.010557891380786895,
+        0.00005552569483406838,
+        0.00008377498388290397,
+        0.0000605235195718706,
+        0.0000366905065602623,
+        0.00005407085546758027,
+        0.00021638836078345788,
+        0.010525680954381826,
+        0.00004830919423839074,
+        0.00007733448530780152,
+        0.00005978569525177588,
+        0.00003123643363360317,
+        0.00004709080595057457,
+        0.0001939378370530904,
+        0.00998856540247798,
+        0.00004645563278463669,
+        0.00007126600248739121,
+        0.00005200477748876437,
+        0.00003055265350267292,
+        0.000045497189089655854,
+        0.00018253272622823706,
+        0.01032583272829652,
+        0.000044604689953848674,
+        0.00006673079421743756,
+        0.000050660950923338505,
+        0.000027101854892680433,
+        0.000045762255322188115,
+        0.00016890384713187808,
+        0.010136223459243775,
+        0.00004455379862338299,
+        0.00006405452857725328,
+        0.00004943669138010592,
+        0.000028590407309820904,
+        0.000039900707779452174,
+        0.00017483096104115243,
+        0.010142471250146626,
+        0.0000416521602543071,
+        0.00006346322163008151,
+        0.00005030547470669263,
+        0.00002662300992524252,
+        0.00003956695257220416,
+        0.00016375841144472352,
+        0.009997803356498476,
+        0.000042872124351561063,
+        0.00006597544132964689,
+        0.00005052208236302249,
+        0.00002720812879269946,
+        0.000039130584619124446,
+        0.0001802312535233797,
+        0.010092107820510864,
+        0.00003863017375115306,
+        0.00006420832830481232,
+        0.000052780072088353346,
+        0.000026064168196171522,
+        0.000040864129550755046,
+        0.00018467701834160846,
+        0.009899575112760066,
+        0.0000431866092141718,
+        0.00006625963510014114,
+        0.00005088201165199277,
+        0.000026399783365195603,
+        0.000037794473674148364,
+        0.00017790157631970968,
+        0.00994323554635048,
+        0.00004512640612665562,
+        0.00007162263081409033,
+        0.00005516039710491889,
+        0.000026367472112178792,
+        0.00004286482669413089,
+        0.0001678395111113784,
+        0.009839660620689389,
+        0.000047105535864830025,
+        0.00007688079774379733,
+        0.00005521634444594385,
+        0.00002967622131109241,
+        0.000041797038167715075,
+        0.00017254053354263292,
+        0.009918793506920338
+      ]
+    }
+  },
+  "sample_ratio": 0.05,
+  "dataset_infos": {
+    "robotgeneralist/pp__1": {
+      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
+      "num_episodes": 237,
+      "num_frames": 99363
+    },
+    "robotgeneralist/pp__1__adversarial": {
+      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
+      "num_episodes": 64,
+      "num_frames": 17906
+    },
+    "robotgeneralist/pp__2": {
+      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
+      "num_episodes": 92,
+      "num_frames": 37458
+    },
+    "robotgeneralist/pp__2__rotations": {
+      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
+      "num_episodes": 102,
+      "num_frames": 13935
+    },
+    "robotgeneralist/pp__3__with_ft": {
+      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
+      "num_episodes": 99,
+      "num_frames": 38424
+    },
+    "robotgeneralist/pp__3__with_ft__rotations": {
+      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
+      "num_episodes": 49,
+      "num_frames": 7151
+    },
+    "robotgeneralist/pp__3__with_ft__adversarial_start": {
+      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
+      "num_episodes": 147,
+      "num_frames": 21313
+    },
+    "robotgeneralist/pp__4__with_ft": {
+      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
+      "num_episodes": 94,
+      "num_frames": 33046
+    },
+    "robotgeneralist/pp__5__with_ft__adversarial_target": {
+      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
+      "num_episodes": 48,
+      "num_frames": 10615
+    },
+    "robotgeneralist/pp__6__with_ft": {
+      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
+      "num_episodes": 379,
+      "num_frames": 123611
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-1": {
+      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
+      "num_episodes": 98,
+      "num_frames": 41410
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-2": {
+      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
+      "num_episodes": 94,
+      "num_frames": 37859
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-3": {
+      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
+      "num_episodes": 127,
+      "num_frames": 53341
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-4": {
+      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
+      "num_episodes": 102,
+      "num_frames": 38263
+    }
+  }
+}

pp__ee_fourier_k50/15000/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}

pp__ee_fourier_k50/15000/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}]]}}"}

pp__ee_fourier_k50/15000/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}}]}

pp__ee_fourier_k50/20000/_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": 1767283006941103548, "commit_timestamp_nsecs": 1767283056409280984, "custom_metadata": {}}

pp__ee_fourier_k50/20000/assets/ur10e/norm_stats.json

@@ -0,0 +1,1525 @@
+{
+  "norm_stats": {
+    "state": {
+      "mean": [
+        0.05313729867339134,
+        0.0037223484832793474,
+        0.106409952044487,
+        0.0008666226058267057,
+        -0.00854077935218811,
+        0.01460699550807476,
+        0.5940247178077698
+      ],
+      "std": [
+        0.1176244392991066,
+        0.16962096095085144,
+        0.07773701101541519,
+        0.04537155479192734,
+        0.06616315245628357,
+        0.6038344502449036,
+        0.49107977747917175
+      ],
+      "q01": [
+        -0.2025573879778385,
+        -0.31106759839057924,
+        0.00890737680345774,
+        -0.11727181684970855,
+        -0.16711108471155167,
+        -1.423702568435669,
+        0.0
+      ],
+      "q99": [
+        0.29901714037060734,
+        0.4115018085956573,
+        0.3217185490995646,
+        0.13774808425903318,
+        0.19241279754638663,
+        1.598234023094177,
+        0.9998
+      ]
+    },
+    "actions": {
+      "mean": [
+        0.0004927651607431471,
+        0.0041174935176968575,
+        0.005378079600632191,
+        -0.0004851108242291957,
+        -0.0010051705176010728,
+        -0.006783927325159311,
+        0.5936462879180908,
+        0.0005593731766566634,
+        -0.00140193453989923,
+        -0.002942675491794944,
+        0.00041861183126457036,
+        0.0004222682327963412,
+        0.0030385160353034735,
+        -0.000329385424265638,
+        -0.0005693831481039524,
+        -0.00029180958517827094,
+        0.000576534541323781,
+        -0.00015302293468266726,
+        -2.4814380594762042e-6,
+        -0.00026009284192696214,
+        0.0003946665092371404,
+        0.000059152920584892854,
+        -0.00006962364568607882,
+        -0.0003498161968309432,
+        0.000053506064432440326,
+        0.00006277022475842386,
+        0.0003592480788938701,
+        -0.00016934213635977358,
+        -0.0001320719748036936,
+        -0.00006360372208291665,
+        0.0001631096820347011,
+        -0.00003750254472834058,
+        -0.000011555765922821593,
+        -0.00006809693149989471,
+        0.00015906375483609736,
+        0.000027621259505394846,
+        -0.00002624911030579824,
+        -0.00012353967758826911,
+        0.000012651072211156134,
+        0.000011669268133118749,
+        0.00012300137314014137,
+        0.00015041981532704085,
+        -0.000056855304137570783,
+        -0.000029746257496299222,
+        0.00007242687570396811,
+        -0.000018986645955010317,
+        -4.556677140499232e-6,
+        -0.00003111416663159616,
+        -0.00009099271846935153,
+        0.000012571626029966865,
+        -0.000013306255823408719,
+        -0.00006410608330043033,
+        6.341869266179856e-6,
+        7.46843352317228e-6,
+        0.00006183652294566855,
+        5.868077096238267e-6,
+        -0.000031927749660098925,
+        -0.00001673644146649167,
+        0.00004087621346116066,
+        -0.000010321637091692537,
+        -2.42824648921669e-6,
+        -0.00001607570084161125,
+        0.00004234428342897445,
+        7.407979410345433e-6,
+        -8.161020559782628e-6,
+        -0.00003870665022986941,
+        3.6879464460071176e-6,
+        4.935619017487625e-6,
+        0.00003818826371571049,
+        0.000051191724196542054,
+        -0.000020595705791492946,
+        -0.000010875675798160955,
+        0.000025950239432859235,
+        -6.903889698151033e-6,
+        -1.2846402341892826e-6,
+        -9.873656381387264e-6,
+        0.00005320658965501934,
+        4.920636911265319e-6,
+        -5.623744073091075e-6,
+        -0.000025824952899711207,
+        2.285133632540237e-6,
+        3.112949343631044e-6,
+        0.00002514909101591911,
+        0.00001734476609271951,
+        -0.000013882496205042116,
+        -7.167188869061647e-6,
+        0.00001790117130440194,
+        -4.3851646296388935e-6,
+        -9.349093375021766e-7,
+        -8.112259820336476e-6,
+        5.336457888915902e-6,
+        3.3939111290237634e-6,
+        -4.019582320324844e-6,
+        -0.00001823278762458358,
+        1.6200752952499897e-6,
+        2.1453445242514135e-6,
+        0.000018565873688203283,
+        0.00004305976108298637,
+        -0.000010276943612552714,
+        -5.3131607273826376e-6,
+        0.000013141003364580683,
+        -3.3397307106497465e-6,
+        -7.699964044149965e-7,
+        -5.316836904967204e-6,
+        -0.00004110661393497139,
+        2.852339093806222e-6,
+        -2.3958800738910213e-6,
+        -0.000013731451872445177,
+        1.5232127452691202e-6,
+        1.386316398566123e-6,
+        0.000013007000234210864,
+        0.00003866312908940017,
+        -7.723581802565604e-6,
+        -3.880812528223032e-6,
+        9.895188668451738e-6,
+        -2.4477972146996763e-6,
+        -5.607461162071559e-7,
+        -3.6636674849432893e-6,
+        0.000010404156455479097,
+        2.0179843431833433e-6,
+        -2.3253292056324426e-6,
+        -0.000010352971003158018,
+        1.0408912203274667e-6,
+        1.1318675205984619e-6,
+        9.978723028325476e-6,
+        -9.606495950720273e-6,
+        -5.949066235189093e-6,
+        -2.8392857984727016e-6,
+        7.4962795224564616e-6,
+        -1.9217382032366004e-6,
+        -4.780694098371896e-7,
+        -3.34773517352005e-6,
+        0.00002107612817781046,
+        1.6454457636427833e-6,
+        -1.817815473259543e-6,
+        -8.123145562421996e-6,
+        7.675727147216094e-7,
+        9.284493671657401e-7,
+        7.868961802159902e-6,
+        -9.203896524923039e-7,
+        -4.736490154755302e-6,
+        -2.4710741399758263e-6,
+        6.142801339592552e-6,
+        -1.6350784335372737e-6,
+        -4.335802543664613e-7,
+        -1.4092432820689282e-6,
+        0.00001656184576859232,
+        1.3327525039130705e-6,
+        -1.3708133792533772e-6,
+        -6.531353847094579e-6,
+        5.90599142924475e-7,
+        6.878227623019484e-7,
+        6.5318749875586946e-6,
+        -4.229010301060043e-6,
+        -4.000812168669654e-6,
+        -2.2653048290521838e-6,
+        5.042587417847244e-6,
+        -1.4821022205069312e-6,
+        -3.451863506143127e-7,
+        -2.389324890827993e-6,
+        8.16222745925188e-6,
+        1.1428018069636892e-6,
+        -1.1332826943544205e-6,
+        -5.504417913471116e-6,
+        5.024350002713618e-7,
+        5.467733217301429e-7,
+        5.279353445075685e-6,
+        -0.0000211214974115137,
+        -3.1678093819209607e-6,
+        -1.8488366322344518e-6,
+        4.2470492189750075e-6,
+        -9.783543646335602e-7,
+        -1.83213785476255e-7,
+        -1.4789864053454949e-6,
+        0.000013087515981169418,
+        7.242887818392774e-7,
+        -1.0046998113466543e-6,
+        -4.340507530287141e-6,
+        3.3989198300332646e-7,
+        5.470611199598352e-7,
+        4.507525773078669e-6,
+        -0.000016270019841613248,
+        -2.829020104400115e-6,
+        -1.296591335631092e-6,
+        3.456672175161657e-6,
+        -9.017520028464787e-7,
+        -1.4440175277741218e-7,
+        -1.4689791214550496e-6,
+        0.000021527563148993067,
+        5.846809472132009e-7,
+        -5.933143256697804e-7,
+        -3.612318550949567e-6,
+        3.5949764765064174e-7,
+        4.328513796281186e-7,
+        3.6886276575387456e-6,
+        -0.000021549723896896467,
+        -2.1144403490325203e-6,
+        -1.4159754755382892e-6,
+        2.68504163614125e-6,
+        -7.679741429456044e-7,
+        -5.776412592695124e-8,
+        -1.192243757941469e-6,
+        0.00003148229370708577,
+        5.478635785038932e-7,
+        -6.972362598389736e-7,
+        -3.211202283637249e-6,
+        2.2451455095051642e-7,
+        3.183014598562295e-7,
+        2.939946398328175e-6,
+        4.565895324049052e-6,
+        -1.7405694734407007e-6,
+        -9.741680742081371e-7,
+        2.2756712496629916e-6,
+        -5.214625389271532e-7,
+        -2.7989250384052866e-7,
+        -8.445824164482474e-7,
+        0.00003105050200247206,
+        5.799903988190636e-7,
+        -7.536694397458632e-7,
+        -2.5344147616124246e-6,
+        2.630972346651106e-7,
+        2.9163163617340615e-7,
+        2.3315690214076312e-6,
+        0.000028919430405949242,
+        -1.4791980902373325e-6,
+        -7.087014637363609e-7,
+        2.1661373921233462e-6,
+        -5.724799052586604e-7,
+        -3.0745661661057966e-7,
+        -5.235916091805848e-7,
+        0.000011437900866440032,
+        3.404710184895521e-7,
+        -4.5958552163938293e-7,
+        -2.1246926280582557e-6,
+        2.5418461291337735e-7,
+        3.871034266467177e-7,
+        1.9572280507418327e-6,
+        0.00003196126999682747,
+        -1.3085544878777e-6,
+        -6.04459614805819e-7,
+        1.575823830535228e-6,
+        -4.1182735799338843e-7,
+        -1.1744142369707333e-7,
+        -1.3322799077286618e-6,
+        2.7163066533830715e-6,
+        3.6782887491426663e-7,
+        -2.6708795530794305e-7,
+        -1.72060231307114e-6,
+        1.8409495794458053e-7,
+        3.7163931665418204e-7,
+        1.8107581354342983e-6,
+        0.000013270215276861563,
+        -1.1510371678014053e-6,
+        -6.630299367316184e-7,
+        1.070535518010729e-6,
+        -3.826233978543314e-7,
+        1.1823525625231923e-7,
+        -1.049707634592778e-6,
+        -2.8137444587628124e-6,
+        2.6974137767865614e-7,
+        -4.0497872078049113e-7,
+        -1.4479496712738182e-6,
+        3.8844518712721765e-8,
+        2.0770157149740953e-8,
+        1.7100152263083146e-6,
+        -0.00002849056909326464,
+        -9.183266342915886e-7,
+        -3.5311214219291287e-7,
+        1.1368990726623451e-6,
+        -3.410139868265105e-7,
+        -2.7979771743957826e-7,
+        -9.036052119881788e-7,
+        -8.269290447060484e-6,
+        3.0624855185124034e-7,
+        -1.1787745535229988e-7,
+        -1.2770473176715313e-6,
+        1.6275393477371836e-7,
+        3.3039643199117563e-7,
+        1.3103665423841449e-6,
+        -0.00001745441295497585,
+        -6.018551061970356e-7,
+        -2.794714930587361e-7,
+        9.792224773264024e-7,
+        -2.040594893060188e-7,
+        -3.4366582468692286e-8,
+        -4.681474479184544e-7,
+        -3.3097076084231958e-6,
+        1.669202376319845e-8,
+        -2.7714148131963157e-7,
+        -9.64512764767278e-7,
+        1.1432685909085194e-7,
+        1.4834664341378812e-7,
+        1.1487189794934238e-6,
+        -0.000010153660696232691,
+        -6.140094228612725e-7,
+        -5.253555173112545e-7,
+        5.710655841539847e-7,
+        -1.404835359153367e-7,
+        4.7323091223461233e-8,
+        -5.46001842849364e-7,
+        -1.2161887070760713e-6,
+        1.2690516371094418e-7,
+        -1.3310798863130913e-7,
+        -7.572793379040377e-7,
+        1.7468086355165724e-7,
+        3.3051062331423964e-8,
+        7.131747565836122e-7,
+        -0.00003459135041339323,
+        -4.08122900807939e-7,
+        -1.8256875478073198e-7,
+        5.756255063715798e-7,
+        -1.6261435575870564e-7,
+        -2.021908862559485e-8,
+        -4.1869603251143417e-7,
+        -1.589809244251228e-6,
+        1.1844038283470582e-7,
+        -7.242162070042468e-8,
+        -4.0011059354583267e-7,
+        -2.4760929306921753e-8,
+        -1.3480507732310798e-7,
+        1.0380339290350094e-6,
+        1.6824413933136384e-6,
+        -4.5796767267347605e-7,
+        -2.0989804738746898e-7,
+        4.763822971654008e-7,
+        -1.7666546625605406e-7,
+        -2.998827852707109e-8,
+        3.5782920804194873e-7,
+        -0.00005497613165061921,
+        -4.8836572119626e-8,
+        -2.8484058134381485e-7,
+        -2.0204961970193835e-7,
+        -1.9280198770843526e-8,
+        8.62186144701127e-9,
+        3.8213460129554733e-7,
+        1.619790509721497e-6,
+        -1.449097339900618e-7,
+        -1.7092831683385157e-7,
+        3.423832595217391e-7,
+        -9.912067611139719e-8,
+        5.862289853553193e-8,
+        -7.740770513464668e-8,
+        -7.812944204488304e-6,
+        1.33551338876714e-7,
+        1.7766551252407226e-7,
+        -2.63828656699161e-8,
+        -5.126612023786947e-9,
+        -4.134697562108158e-8,
+        8.136574791706153e-8,
+        -4.875783361057984e-6
+      ],
+      "std": [
+        0.054471205919981,
+        0.07741593569517136,
+        0.06635452806949615,
+        0.02219589799642563,
+        0.03713744506239891,
+        0.2155904769897461,
+        0.44884487986564636,
+        0.022129571065306664,
+        0.030908629298210144,
+        0.027230359613895416,
+        0.009623671881854534,
+        0.015185607597231865,
+        0.08782459795475006,
+        0.11000800132751465,
+        0.0054918802343308926,
+        0.007706818170845509,
+        0.007092009298503399,
+        0.004339421167969704,
+        0.006056886166334152,
+        0.025186194106936455,
+        0.0549297109246254,
+        0.0023341968189924955,
+        0.003104041563346982,
+        0.0028293700888752937,
+        0.002068394096568227,
+        0.0026343404315412045,
+        0.01149396225810051,
+        0.03640395402908325,
+        0.0011078902753069997,
+        0.0015103736659511924,
+        0.0014142218278720975,
+        0.0010499481577426195,
+        0.0013543356908485293,
+        0.005356063600629568,
+        0.027538813650608063,
+        0.0007573548937216401,
+        0.001025661244057119,
+        0.0009198821499012411,
+        0.0005544671439565718,
+        0.0007547291461378336,
+        0.0035150113981217146,
+        0.02213539369404316,
+        0.000455506524303928,
+        0.0006300596869550645,
+        0.0005910721374675632,
+        0.00034599038190208375,
+        0.0004919870407320559,
+        0.002043767599388957,
+        0.018490547314286232,
+        0.00037283002166077495,
+        0.0005109719350002706,
+        0.00045568859786726534,
+        0.00023278511071112007,
+        0.00032812528661452234,
+        0.0016570856096222997,
+        0.015843955799937248,
+        0.0002481868432369083,
+        0.0003480722662061453,
+        0.000325335975503549,
+        0.00017280328029301018,
+        0.0002512190258130431,
+        0.0010802638716995716,
+        0.01408928632736206,
+        0.00022223185806069523,
+        0.00030612919363193214,
+        0.0002718937466852367,
+        0.00012990496179554611,
+        0.00018422518041916192,
+        0.000973312184214592,
+        0.012508493848145008,
+        0.00015761872055009007,
+        0.00022044251090846956,
+        0.0002061627892544493,
+        0.0001057182380463928,
+        0.00015497820277232677,
+        0.0006764288409613073,
+        0.011296260170638561,
+        0.00014791855937801301,
+        0.00020399357890710235,
+        0.00018064411415252835,
+        0.00008491610060445964,
+        0.0001210872214869596,
+        0.0006432573427446187,
+        0.01023485790938139,
+        0.00010906582610914484,
+        0.00015326884749811143,
+        0.000142823948408477,
+        0.00007245635788422078,
+        0.00010654709330992773,
+        0.00046615453902632,
+        0.009492816403508186,
+        0.00010573185136308894,
+        0.0001460392086300999,
+        0.0001292345259571448,
+        0.00006049493094906211,
+        0.00008650264499010518,
+        0.00045887212036177516,
+        0.008779959753155708,
+        0.00007973915489856154,
+        0.00011203208850929514,
+        0.00010434568685013801,
+        0.00005302290082909167,
+        0.00007767079659970477,
+        0.0003416067629586905,
+        0.00814682524651289,
+        0.00007915469905128703,
+        0.00010886993550229818,
+        0.0000966577063081786,
+        0.000045318782213144004,
+        0.00006455113179981709,
+        0.0003425226896069944,
+        0.007630111649632454,
+        0.0000611644281889312,
+        0.00008592260564910248,
+        0.00007935166649986058,
+        0.000040576029277872294,
+        0.00005935502122156322,
+        0.00026016749325208366,
+        0.007179272826761007,
+        0.00006150962872197852,
+        0.00008518873801222071,
+        0.00007461248605977744,
+        0.0000355022020812612,
+        0.0000502788316225633,
+        0.00026474526384845376,
+        0.006768743507564068,
+        0.000048591213271720335,
+        0.00006827081961091608,
+        0.00006262768874876201,
+        0.000032297353754984215,
+        0.000046926012146286666,
+        0.00020619359565898776,
+        0.006469320971518755,
+        0.00004949716094415635,
+        0.00006947917427169159,
+        0.00005963100556982681,
+        0.000028536098398035392,
+        0.000040751590859144926,
+        0.00021295368787832558,
+        0.006215229630470276,
+        0.00003966964141000062,
+        0.00005751234493800439,
+        0.00005083374344394542,
+        0.00002690069413802121,
+        0.00003854740134556778,
+        0.00016899738693609834,
+        0.005915363319218159,
+        0.00004071096918778494,
+        0.00005744508234784007,
+        0.000049411490181228146,
+        0.000024393977582803927,
+        0.000034359978599241003,
+        0.00017412411398254335,
+        0.005679433699697256,
+        0.000034098236938007176,
+        0.00004947670822730288,
+        0.0000428831554017961,
+        0.000023318862076848745,
+        0.0000333583120664116,
+        0.00014037704386282712,
+        0.005502832122147083,
+        0.000035476441553328186,
+        0.00005029448220739141,
+        0.000041998053347924724,
+        0.000021735353584517725,
+        0.00003051272506127134,
+        0.00014704464410897344,
+        0.005227266810834408,
+        0.000031422849133377895,
+        0.0000454086039098911,
+        0.000037437119317473844,
+        0.000021104986444697715,
+        0.0000290283169306349,
+        0.00012362531560938805,
+        0.005091419909149408,
+        0.00003195118915755302,
+        0.00004717931369668804,
+        0.00003730821481440216,
+        0.000018998216546606272,
+        0.00002651557770150248,
+        0.0001294043322559446,
+        0.004971646703779697,
+        0.00002650190617714543,
+        0.000040163333324017,
+        0.00003248209395678714,
+        0.000018027179976343177,
+        0.00002582307206466794,
+        0.00010801247117342427,
+        0.004754771012812853,
+        0.00002661882717802655,
+        0.00003864263999275863,
+        0.00003127731179120019,
+        0.000016530248103663325,
+        0.000023721948309685104,
+        0.00010995724005624652,
+        0.004578577820211649,
+        0.000022862250261823647,
+        0.000033582466130610555,
+        0.000027823365599033423,
+        0.000015542551409453154,
+        0.000022585352780879475,
+        0.00009167923417408019,
+        0.004508121404796839,
+        0.00002280540320498403,
+        0.000033627373341005296,
+        0.00002707460953388363,
+        0.000014449000445893034,
+        0.00002091616806865204,
+        0.0000951679321588017,
+        0.004384348168969154,
+        0.00002045510154857766,
+        0.00002992084773723036,
+        0.000024755014237598516,
+        0.000014510045730276033,
+        0.000020438790670596063,
+        0.00008195763803087175,
+        0.004295016173273325,
+        0.000021367646695580333,
+        0.00003022674354724586,
+        0.000024624087018310092,
+        0.000013480367670126725,
+        0.000019035556761082262,
+        0.0000849158750497736,
+        0.004229281097650528,
+        0.000018941680536954664,
+        0.000027560265152715147,
+        0.000022634478227701038,
+        0.000013048511391389184,
+        0.00001906735087686684,
+        0.00007548843132099137,
+        0.004119527991861105,
+        0.00002021879299718421,
+        0.00002808637873386033,
+        0.000022752908989787102,
+        0.000012890232028439641,
+        0.000019045859517063946,
+        0.0000775336375227198,
+        0.004028849303722382,
+        0.00001860819975263439,
+        0.000026566647647996433,
+        0.000021319850930012763,
+        0.000012895960026071407,
+        0.00001921564035001211,
+        0.0000719531744834967,
+        0.003980362322181463,
+        0.000019542083464330062,
+        0.000027795384085038677,
+        0.000021858841137145646,
+        0.000013172734725230839,
+        0.000019754204913624562,
+        0.00007459476182702929,
+        0.003902778960764408,
+        0.00002043036147370003,
+        0.00003195027602487244,
+        0.000024560755264246836,
+        0.000014010387531016022,
+        0.00002065290027530864,
+        0.00007676216046093032,
+        0.0038293309044092894,
+        0.000023211092411656864,
+        0.00003743018169188872,
+        0.000025842035029199906,
+        0.000015010549759608693,
+        0.000020894838598906063,
+        0.00008554229862056673,
+        0.0037848083302378654,
+        0.00002203849180659745,
+        0.00003869964712066576,
+        0.00002533109909563791,
+        0.000013811703865940217,
+        0.000020183759261271916,
+        0.0000796892709331587,
+        0.0037312512286007404,
+        0.000020139470507274382,
+        0.00003185970854246989,
+        0.00002326746835024096,
+        0.000014411976735573262,
+        0.000020371249775053002,
+        0.00007360208837781101,
+        0.0036642765626311302,
+        0.000017049775124178268,
+        0.000027688640329870395,
+        0.00001989392148971092,
+        0.000012698577847913839,
+        0.000017972832210944034,
+        0.00006560993642779067,
+        0.0036550285294651985,
+        0.000016028958270908333,
+        0.000024884970116545446,
+        0.00001818954842747189,
+        0.000011278674719505943,
+        0.000015812189303687774,
+        0.00006195339665282518,
+        0.0036316923797130585,
+        0.00001560178498039022,
+        0.000024082159143290482,
+        0.00001761745625117328,
+        0.00001089257057174109,
+        0.000015621577404090203,
+        0.00005938451431575231,
+        0.003624820150434971,
+        0.000014879788068355992,
+        0.00002372561357333325,
+        0.00001737473940011114,
+        0.00001018085913528921,
+        0.000015368110325653106,
+        0.00005791655712528154,
+        0.0035625966265797615,
+        0.000014492210539174266,
+        0.000023905164198367856,
+        0.000016788300854386762,
+        0.00001016644364426611,
+        0.000014378600099007599,
+        0.00005657917427015491,
+        0.0035458113998174667,
+        0.000014646487215941306,
+        0.000023379501726594754,
+        0.000016420719475718215,
+        0.000010098652637680061,
+        0.000014440963241213467,
+        0.00005597082417807542,
+        0.0035429371055215597,
+        0.000014022874893271364,
+        0.00002083050821966026,
+        0.000015985659047146328,
+        0.00001016553051158553,
+        0.000013916131138103083,
+        0.000056019474868662655,
+        0.0034998103510588408,
+        0.000014245401871448848,
+        0.000022099453417467885,
+        0.00001657450047787279,
+        0.000010621713045111392,
+        0.000014651923265773803,
+        0.00005747186514781788,
+        0.0034743566066026688,
+        0.000014478311641141772,
+        0.00002349081842112355,
+        0.000017406200640834868,
+        0.000010810344974743202,
+        0.000014864897821098566,
+        0.00005688491728506051,
+        0.0034770709462463856,
+        0.000016515868992428295,
+        0.00002587365270301234,
+        0.000018931375961983576,
+        0.000010814174856932368,
+        0.000016132373275468126,
+        0.00006046127964509651,
+        0.003478288883343339
+      ],
+      "q01": [
+        -0.15909142580032348,
+        -0.23591104531884194,
+        -0.1173719973474741,
+        -0.0725411132991314,
+        -0.12759929532110692,
+        -0.6792203711509704,
+        0.0,
+        -0.06588815684318543,
+        -0.1043403072744608,
+        -0.07569241938591004,
+        -0.026123490297794344,
+        -0.04355562012195587,
+        -0.26860033273696904,
+        -0.31594268302917483,
+        -0.017998227071762084,
+        -0.025448361220955847,
+        -0.021794256627559662,
+        -0.014516740383952856,
+        -0.017180888187885286,
+        -0.08126688668131828,
+        -0.15649057692885399,
+        -0.006211889632977545,
+        -0.009136751361563802,
+        -0.00862356810811907,
+        -0.00636542599722743,
+        -0.007391496415808798,
+        -0.035477512252330784,
+        -0.10543232785761356,
+        -0.003451917093619704,
+        -0.004779039949178695,
+        -0.00400189593732357,
+        -0.003541165339201688,
+        -0.003899161383137107,
+        -0.016611159443855286,
+        -0.07849292940199376,
+        -0.001959954298567027,
+        -0.0028813693393021818,
+        -0.0027899963006842885,
+        -0.0015735435530543325,
+        -0.002058688412979245,
+        -0.010417635813355447,
+        -0.0639662309139967,
+        -0.0013655057244002819,
+        -0.0019750579357147215,
+        -0.0016051418443210422,
+        -0.0011298993272706869,
+        -0.0014582404736429451,
+        -0.006558569058775902,
+        -0.05251251416504383,
+        -0.0009643727889051661,
+        -0.0014315907768905163,
+        -0.001390606227517128,
+        -0.0006525630556046963,
+        -0.0008888287223875523,
+        -0.004895182948186994,
+        -0.0455008171916008,
+        -0.0007486765272915363,
+        -0.0010981693975627422,
+        -0.000882065705396235,
+        -0.0005725251567782834,
+        -0.0007578355811536312,
+        -0.0034509299120865766,
+        -0.04015725544691086,
+        -0.0005749710650416091,
+        -0.0008555282396264373,
+        -0.0008345671106129884,
+        -0.0003548953181132674,
+        -0.0004969418092630804,
+        -0.0028254041438922285,
+        -0.035569155138731004,
+        -0.0004705392210977152,
+        -0.0006840464788489043,
+        -0.0005612756710499526,
+        -0.00035702317333780227,
+        -0.0004648537166416645,
+        -0.002128881996870041,
+        -0.03083838468044997,
+        -0.00038048800966935235,
+        -0.000566111233481206,
+        -0.0005538229705183766,
+        -0.00023378437627106905,
+        -0.00032636490603908896,
+        -0.0018706658044829966,
+        -0.02931403119564057,
+        -0.00032479510342236606,
+        -0.000475912622641772,
+        -0.00038455819835653526,
+        -0.00024285348085686562,
+        -0.0003135723091545515,
+        -0.001452100250311196,
+        -0.026792802512645722,
+        -0.00027214939426630733,
+        -0.00040756029058247796,
+        -0.0003960524151916616,
+        -0.00016536210232879968,
+        -0.00023339646961539982,
+        -0.0013170160830486568,
+        -0.02499642580747604,
+        -0.00024108570310054343,
+        -0.000348888192162849,
+        -0.0002832006874959916,
+        -0.00017626483961939813,
+        -0.00023037131156306717,
+        -0.0010707858602516352,
+        -0.023444418999552727,
+        -0.00020677300081588327,
+        -0.0003006568572251126,
+        -0.0002946901318966411,
+        -0.00012258605094393713,
+        -0.00017606193688698117,
+        -0.0009890523520298302,
+        -0.02203794540464878,
+        -0.0001841578755469527,
+        -0.00026803012369200585,
+        -0.00021574946893379094,
+        -0.00013577964630676436,
+        -0.0001734850220615044,
+        -0.0008068450814578683,
+        -0.020441363253444433,
+        -0.00016045538827893325,
+        -0.000241462552268058,
+        -0.00022808200246654452,
+        -0.00009496491550235078,
+        -0.00013633702206425368,
+        -0.0007666170152835548,
+        -0.019491385824978353,
+        -0.0001477667265571654,
+        -0.00021141915648477145,
+        -0.0001735861329594627,
+        -0.00010635797707363962,
+        -0.00013736910806037486,
+        -0.0006427704328671098,
+        -0.018371846283972263,
+        -0.0001293026403174736,
+        -0.00019473347970051693,
+        -0.0001822685490187723,
+        -0.00007863951541366989,
+        -0.00010972291373182087,
+        -0.0006055113250389692,
+        -0.017673968444019555,
+        -0.00011848664400167762,
+        -0.00017848168123746295,
+        -0.00014009435730986296,
+        -0.0000859230127884075,
+        -0.00011260414165444673,
+        -0.0005137789773289115,
+        -0.016190811094641686,
+        -0.00010690572904422879,
+        -0.0001638943360070698,
+        -0.0001501999743981287,
+        -0.00006396331703872426,
+        -0.00009358972666086629,
+        -0.0005007586104329676,
+        -0.01619275164604187,
+        -0.00010315810496686024,
+        -0.00015018478848505767,
+        -0.00011576190601917915,
+        -0.00007506265956908466,
+        -0.00009504533028230073,
+        -0.0004348065205849708,
+        -0.01565815359354019,
+        -0.00009185501914471398,
+        -0.0001392430538544431,
+        -0.00012555517936125394,
+        -0.00005638334401883189,
+        -0.00008002441792050375,
+        -0.000424562167050317,
+        -0.015019814673811197,
+        -0.00009095547425094991,
+        -0.00013837829248514026,
+        -0.00010207304285722785,
+        -0.00006334332783007992,
+        -0.00008174893613904723,
+        -0.0003743791728978976,
+        -0.014377124969661235,
+        -0.00008569652531296014,
+        -0.00013066508993506436,
+        -0.00010974786025471983,
+        -0.000049689067096915106,
+        -0.00007135935623664408,
+        -0.0003742531020194291,
+        -0.014187388128042218,
+        -0.0000794089021859691,
+        -0.00011592403496615584,
+        -0.0000893614823697135,
+        -0.0000560631047235802,
+        -0.0000719853327609599,
+        -0.0003315736224874853,
+        -0.013546264378726483,
+        -0.0000717077999375761,
+        -0.00010777361229993398,
+        -0.0000922710225568153,
+        -0.00004497638295288198,
+        -0.00006191108026541768,
+        -0.0003081054835114628,
+        -0.013243352345749734,
+        -0.00006864526765421036,
+        -0.0001027144630090333,
+        -0.00008054810634930621,
+        -0.000049568866193294524,
+        -0.00006090367061551658,
+        -0.00028183929068036377,
+        -0.012778499507904054,
+        -0.00006117501191329209,
+        -0.00009796037861378861,
+        -0.00008128996908781118,
+        -0.00004006202814634888,
+        -0.00005725241685286168,
+        -0.0002700211696093902,
+        -0.012565325374901296,
+        -0.00006034678600262851,
+        -0.0000907877254066989,
+        -0.00007010409524082204,
+        -0.000041811013757251194,
+        -0.0000556492382194847,
+        -0.0002493678867351264,
+        -0.011774876400828364,
+        -0.00005579026520717887,
+        -0.00008717787461355332,
+        -0.00007443824684014546,
+        -0.0000356287220492959,
+        -0.00005372272633831019,
+        -0.00024090840241406124,
+        -0.011997990143299102,
+        -0.000058186027396004656,
+        -0.00008279621332185345,
+        -0.00006459413380362097,
+        -0.00003831381897907702,
+        -0.00005259611281217076,
+        -0.00022310287454165522,
+        -0.011667696542292834,
+        -0.00005284995273686948,
+        -0.00008384203007444737,
+        -0.00006744749160716312,
+        -0.000032252046390203774,
+        -0.0000481185418262612,
+        -0.00021629847201984375,
+        -0.011543062198162084,
+        -0.00005847320245811716,
+        -0.00008175599463284015,
+        -0.00006548594534397124,
+        -0.000038102848210837704,
+        -0.0000492544468608685,
+        -0.00021983185317367318,
+        -0.011402166564017536,
+        -0.00005387163197156041,
+        -0.0000825874566333369,
+        -0.00006687336476170457,
+        -0.000033979644608916735,
+        -0.00005101760845864191,
+        -0.00020870170197449626,
+        -0.011224610126763582,
+        -0.00006089058108045721,
+        -0.00008930324914399536,
+        -0.00007119371397420766,
+        -0.00003768286973936482,
+        -0.000053939796844497314,
+        -0.0002299774368293584,
+        -0.010891138190776108,
+        -0.000060360829543788016,
+        -0.00009397184122353786,
+        -0.00007481435872614382,
+        -0.00003693910902366043,
+        -0.00005306754379416818,
+        -0.0002138554411940277,
+        -0.01082267729640007,
+        -0.00005871810513781385,
+        -0.00008991035101935281,
+        -0.00007033640673034824,
+        -0.000036662435624748536,
+        -0.00005430486062541601,
+        -0.0002200629871338608,
+        -0.010579705061763527,
+        -0.000053560086898505666,
+        -0.00008582322522997861,
+        -0.00007200759416446087,
+        -0.000034787034674081956,
+        -0.000048553584120236346,
+        -0.0002130190141499044,
+        -0.010566252829506993,
+        -0.00005062428400851786,
+        -0.00007914942941861225,
+        -0.000059331482945708576,
+        -0.00003430225813062863,
+        -0.00004731270743068308,
+        -0.00019290218278765676,
+        -0.010022874112427235,
+        -0.000047412463498767456,
+        -0.00007506768405437467,
+        -0.00005783305459190161,
+        -0.00002997652450576426,
+        -0.00004301143199554646,
+        -0.00018155146474018715,
+        -0.010345511507242918,
+        -0.00004604501994326713,
+        -0.00007326857978478065,
+        -0.00005605542755220084,
+        -0.000028739307308569553,
+        -0.00004117103526368737,
+        -0.0001768201191909612,
+        -0.010309934300184249,
+        -0.00004275882025249304,
+        -0.00006731156632304187,
+        -0.00005409912285394967,
+        -0.000026329884584993103,
+        -0.00004337953124195339,
+        -0.00016598801976069823,
+        -0.01018075596317649,
+        -0.000043296541273593885,
+        -0.00006625080597586939,
+        -0.00005305007540155199,
+        -0.000028063857986126094,
+        -0.000040029496920760744,
+        -0.00017517980523407465,
+        -0.010162209048867225,
+        -0.0000426195073465351,
+        -0.00006396621902240442,
+        -0.000051819634152343525,
+        -0.000029033373587299122,
+        -0.00004320319109829139,
+        -0.00016554977782070636,
+        -0.010125353442132473,
+        -0.000045235728612169645,
+        -0.00006515037268400192,
+        -0.00004881905938964337,
+        -0.000028501104679889977,
+        -0.00004041612264700232,
+        -0.00016903663412667815,
+        -0.010062240642309189,
+        -0.000043975374847650503,
+        -0.00006714530037716033,
+        -0.00005093284538015727,
+        -0.00002801080799545163,
+        -0.00004157804111018775,
+        -0.00016746679800562535,
+        -0.009975232541561126,
+        -0.000045126923860516424,
+        -0.00007225614520721142,
+        -0.00005288218818604951,
+        -0.000029170855134725583,
+        -0.000040984369814395916,
+        -0.0001663898449391129,
+        -0.009851353891193867,
+        -0.00004605035483837126,
+        -0.00007112415134906766,
+        -0.0000582737371325493,
+        -0.000028596166521310774,
+        -0.000044461411982774806,
+        -0.0001793744623661041,
+        -0.009932486386597156
+      ],
+      "q99": [
+        0.16212226042747496,
+        0.25229084351658815,
+        0.18486946655809877,
+        0.061344154804944984,
+        0.10886132830679413,
+        0.6657196632862092,
+        0.9998,
+        0.06576045138835906,
+        0.09603128664791583,
+        0.045872775423526774,
+        0.03236173896193503,
+        0.05487722767591477,
+        0.2764705369472503,
+        0.3158153381347656,
+        0.014891946853697303,
+        0.023800758841633796,
+        0.018585946798324586,
+        0.012693745650351043,
+        0.021670551879704,
+        0.08171248322725297,
+        0.15637542012631894,
+        0.0069066042726859435,
+        0.009335029612295327,
+        0.0047039436949416966,
+        0.006996506480127576,
+        0.008702572771534321,
+        0.036967494320869446,
+        0.1053792189329863,
+        0.002857540221139789,
+        0.004296822899580001,
+        0.003630450583249331,
+        0.003087930056080223,
+        0.004339818637073041,
+        0.01760330287218094,
+        0.07830542102754112,
+        0.0021181412500329315,
+        0.0029414672447368503,
+        0.0015112007031682874,
+        0.0019269651249051098,
+        0.0024494722973555337,
+        0.01144827667623758,
+        0.06380807581543924,
+        0.0011728783883154393,
+        0.001813931218627841,
+        0.0014501392262987797,
+        0.0009983487627469,
+        0.0015205995496362446,
+        0.006439514055848122,
+        0.05315425983667374,
+        0.0010328240715665742,
+        0.0014663059995509691,
+        0.0007404281431809068,
+        0.0008001560054719447,
+        0.001095746040716767,
+        0.0052222565246745935,
+        0.04538371631056071,
+        0.0006383903683163227,
+        0.0009848102391697465,
+        0.000784420038573444,
+        0.0004887675133766608,
+        0.0007893593505024911,
+        0.0033699059166945513,
+        0.039476715770363824,
+        0.0006163579971762373,
+        0.0008800703473389147,
+        0.00044086696542799475,
+        0.000447366009419784,
+        0.0006247742644743994,
+        0.0030928143300116076,
+        0.035530846929550164,
+        0.0004072732983855529,
+        0.0006279987238347533,
+        0.0004935565162450075,
+        0.0003003585597965866,
+        0.0004761313281953334,
+        0.002095991748571396,
+        0.030764527727663513,
+        0.00040901676943758494,
+        0.0005783375784289091,
+        0.00029091126989806075,
+        0.0002924530396237969,
+        0.0004053453548112883,
+        0.0020267377858981485,
+        0.02924695993065833,
+        0.0002835550721851177,
+        0.0004330007825046778,
+        0.0003438553759129719,
+        0.00021280083870515228,
+        0.0003332204633043148,
+        0.0014287737663835282,
+        0.02666576117277146,
+        0.00029395890273153794,
+        0.00041647118404507634,
+        0.0002144097874115686,
+        0.00020652228510007256,
+        0.0002845425325096586,
+        0.0014505576011724767,
+        0.024956510066986085,
+        0.00020434754303423695,
+        0.00031225532339885847,
+        0.0002542992958333343,
+        0.0001512215134222061,
+        0.00024444468863075595,
+        0.0010491569862701002,
+        0.023069663181155914,
+        0.00022068456518463788,
+        0.00031388382902368915,
+        0.00016011388470651587,
+        0.00015496468472993004,
+        0.00021093179946765303,
+        0.001086882085958496,
+        0.0219863745495677,
+        0.00016183016332215621,
+        0.00024126361515372995,
+        0.0001919538950081915,
+        0.00011462044673971831,
+        0.00018409015340730545,
+        0.0008034803080838172,
+        0.020384925924241543,
+        0.0001725941591721494,
+        0.00024417606623610485,
+        0.00012369809069205076,
+        0.00012043028190964833,
+        0.00016321412267861885,
+        0.0008335524674970658,
+        0.019483701366186136,
+        0.00012494574459269637,
+        0.0001922543402411975,
+        0.00015176694963593034,
+        0.00008988425261923113,
+        0.00014705147650092838,
+        0.0006307619461789725,
+        0.01831897035390139,
+        0.00013962995521724217,
+        0.00019706185359973463,
+        0.00010126548151019962,
+        0.0000950566966785118,
+        0.00013232208000263205,
+        0.000660218992456794,
+        0.01764387315139175,
+        0.00010247177979908883,
+        0.00015943087645573536,
+        0.00012344940464245158,
+        0.00007527396054356363,
+        0.00011956113649066536,
+        0.0005160252890083939,
+        0.016168088349699967,
+        0.00011438966927817092,
+        0.00016185711201978847,
+        0.00008490921122720466,
+        0.00007747934755752788,
+        0.00010678794741397725,
+        0.0005481199477799237,
+        0.01615487493276596,
+        0.00008468688050634237,
+        0.00013357372635509814,
+        0.00010555801286245698,
+        0.00006330196186900142,
+        0.000101998839713633,
+        0.00042467831117101035,
+        0.015395162180066112,
+        0.00009998356355354183,
+        0.00014026010758243506,
+        0.00007274458486353978,
+        0.00006777426579501484,
+        0.00009303196137771012,
+        0.0004640237890882414,
+        0.01499328883215785,
+        0.00007716326334048071,
+        0.00011421011368511244,
+        0.00009314750883495434,
+        0.00005738810400944202,
+        0.00008783590141683817,
+        0.0003717806453816595,
+        0.014554040935635573,
+        0.00008682058225385845,
+        0.00012317761406302453,
+        0.00007470068163238468,
+        0.00005833794826176018,
+        0.0000807305663009174,
+        0.0003966953692026436,
+        0.014136480166763069,
+        0.00006818514921469611,
+        0.00010206100093200813,
+        0.00008256861115805807,
+        0.000049349277792498495,
+        0.00007737125915009528,
+        0.00031947106840088976,
+        0.013474849896132943,
+        0.00007525306886527682,
+        0.00010829368010163308,
+        0.00006185477034305223,
+        0.000050571078446228043,
+        0.00007023279787972571,
+        0.0003371094720438124,
+        0.013198494754731657,
+        0.000059138852264732124,
+        0.00008794275637483227,
+        0.00007062362715369087,
+        0.00004125927500426771,
+        0.00006884480739245192,
+        0.0002681864194571972,
+        0.012739881089329719,
+        0.00006564935145433995,
+        0.00009313276901375502,
+        0.00005367935128742829,
+        0.00004326853083912283,
+        0.000059486858663149135,
+        0.000290332669764757,
+        0.012545088732242582,
+        0.00005332418816396964,
+        0.0000798318615881726,
+        0.00006209178492426877,
+        0.00003812435691943395,
+        0.00005920125464908778,
+        0.00024440587060526003,
+        0.011752966091036793,
+        0.00006044615641003474,
+        0.00008350273473188281,
+        0.00005087533958721905,
+        0.00004026508983224629,
+        0.00005588843687437471,
+        0.0002512834585038945,
+        0.01198878407478332,
+        0.00004965451551834123,
+        0.00007573206002125512,
+        0.00006231120838783678,
+        0.00003364615770988169,
+        0.00005282912561669945,
+        0.0002209204529877752,
+        0.011624232469499118,
+        0.00005783643363974989,
+        0.00008142779953777802,
+        0.00005095888577052386,
+        0.000036499108670977874,
+        0.000052707964507862936,
+        0.000231120909564197,
+        0.01148185554742813,
+        0.00004983878775965424,
+        0.00007604824067093425,
+        0.000057698829320725065,
+        0.000033903982862830136,
+        0.000054097583203110825,
+        0.00020294103957712662,
+        0.01119744649901986,
+        0.00005782577866921202,
+        0.00007854039029916743,
+        0.0000542719262011815,
+        0.00003704023372847582,
+        0.00005206325108883899,
+        0.00022946770491544147,
+        0.011211802294850345,
+        0.0000541387221601326,
+        0.00008076134007424106,
+        0.00006221706023206938,
+        0.00003468251589220018,
+        0.000056180346361361434,
+        0.00020245690247975277,
+        0.010855805838853122,
+        0.00005655899195699014,
+        0.0000880048372084274,
+        0.00006508456927258521,
+        0.00003773696059361102,
+        0.00005296763135702342,
+        0.00022277740947902205,
+        0.010783490767329935,
+        0.00005639911381294957,
+        0.00008836033511906856,
+        0.00007074359970283693,
+        0.00003449545800685873,
+        0.00004960680939257152,
+        0.0002082344447262587,
+        0.010557891380786895,
+        0.00005552569483406838,
+        0.00008377498388290397,
+        0.0000605235195718706,
+        0.0000366905065602623,
+        0.00005407085546758027,
+        0.00021638836078345788,
+        0.010525680954381826,
+        0.00004830919423839074,
+        0.00007733448530780152,
+        0.00005978569525177588,
+        0.00003123643363360317,
+        0.00004709080595057457,
+        0.0001939378370530904,
+        0.00998856540247798,
+        0.00004645563278463669,
+        0.00007126600248739121,
+        0.00005200477748876437,
+        0.00003055265350267292,
+        0.000045497189089655854,
+        0.00018253272622823706,
+        0.01032583272829652,
+        0.000044604689953848674,
+        0.00006673079421743756,
+        0.000050660950923338505,
+        0.000027101854892680433,
+        0.000045762255322188115,
+        0.00016890384713187808,
+        0.010136223459243775,
+        0.00004455379862338299,
+        0.00006405452857725328,
+        0.00004943669138010592,
+        0.000028590407309820904,
+        0.000039900707779452174,
+        0.00017483096104115243,
+        0.010142471250146626,
+        0.0000416521602543071,
+        0.00006346322163008151,
+        0.00005030547470669263,
+        0.00002662300992524252,
+        0.00003956695257220416,
+        0.00016375841144472352,
+        0.009997803356498476,
+        0.000042872124351561063,
+        0.00006597544132964689,
+        0.00005052208236302249,
+        0.00002720812879269946,
+        0.000039130584619124446,
+        0.0001802312535233797,
+        0.010092107820510864,
+        0.00003863017375115306,
+        0.00006420832830481232,
+        0.000052780072088353346,
+        0.000026064168196171522,
+        0.000040864129550755046,
+        0.00018467701834160846,
+        0.009899575112760066,
+        0.0000431866092141718,
+        0.00006625963510014114,
+        0.00005088201165199277,
+        0.000026399783365195603,
+        0.000037794473674148364,
+        0.00017790157631970968,
+        0.00994323554635048,
+        0.00004512640612665562,
+        0.00007162263081409033,
+        0.00005516039710491889,
+        0.000026367472112178792,
+        0.00004286482669413089,
+        0.0001678395111113784,
+        0.009839660620689389,
+        0.000047105535864830025,
+        0.00007688079774379733,
+        0.00005521634444594385,
+        0.00002967622131109241,
+        0.000041797038167715075,
+        0.00017254053354263292,
+        0.009918793506920338
+      ]
+    }
+  },
+  "sample_ratio": 0.05,
+  "dataset_infos": {
+    "robotgeneralist/pp__1": {
+      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
+      "num_episodes": 237,
+      "num_frames": 99363
+    },
+    "robotgeneralist/pp__1__adversarial": {
+      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
+      "num_episodes": 64,
+      "num_frames": 17906
+    },
+    "robotgeneralist/pp__2": {
+      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
+      "num_episodes": 92,
+      "num_frames": 37458
+    },
+    "robotgeneralist/pp__2__rotations": {
+      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
+      "num_episodes": 102,
+      "num_frames": 13935
+    },
+    "robotgeneralist/pp__3__with_ft": {
+      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
+      "num_episodes": 99,
+      "num_frames": 38424
+    },
+    "robotgeneralist/pp__3__with_ft__rotations": {
+      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
+      "num_episodes": 49,
+      "num_frames": 7151
+    },
+    "robotgeneralist/pp__3__with_ft__adversarial_start": {
+      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
+      "num_episodes": 147,
+      "num_frames": 21313
+    },
+    "robotgeneralist/pp__4__with_ft": {
+      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
+      "num_episodes": 94,
+      "num_frames": 33046
+    },
+    "robotgeneralist/pp__5__with_ft__adversarial_target": {
+      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
+      "num_episodes": 48,
+      "num_frames": 10615
+    },
+    "robotgeneralist/pp__6__with_ft": {
+      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
+      "num_episodes": 379,
+      "num_frames": 123611
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-1": {
+      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
+      "num_episodes": 98,
+      "num_frames": 41410
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-2": {
+      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
+      "num_episodes": 94,
+      "num_frames": 37859
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-3": {
+      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
+      "num_episodes": 127,
+      "num_frames": 53341
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-4": {
+      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
+      "num_episodes": 102,
+      "num_frames": 38263
+    }
+  }
+}

pp__ee_fourier_k50/20000/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}

pp__ee_fourier_k50/20000/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}]]}}"}

pp__ee_fourier_k50/20000/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}}]}

pp__ee_fourier_k50/25000/_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": 1767292665108226660, "commit_timestamp_nsecs": 1767292706964266079, "custom_metadata": {}}

pp__ee_fourier_k50/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": 1767254079621082034, "commit_timestamp_nsecs": 1767254128354000700, "custom_metadata": {}}

pp__ee_fourier_k50/5000/assets/ur10e/norm_stats.json

@@ -0,0 +1,1525 @@
+{
+  "norm_stats": {
+    "state": {
+      "mean": [
+        0.05313729867339134,
+        0.0037223484832793474,
+        0.106409952044487,
+        0.0008666226058267057,
+        -0.00854077935218811,
+        0.01460699550807476,
+        0.5940247178077698
+      ],
+      "std": [
+        0.1176244392991066,
+        0.16962096095085144,
+        0.07773701101541519,
+        0.04537155479192734,
+        0.06616315245628357,
+        0.6038344502449036,
+        0.49107977747917175
+      ],
+      "q01": [
+        -0.2025573879778385,
+        -0.31106759839057924,
+        0.00890737680345774,
+        -0.11727181684970855,
+        -0.16711108471155167,
+        -1.423702568435669,
+        0.0
+      ],
+      "q99": [
+        0.29901714037060734,
+        0.4115018085956573,
+        0.3217185490995646,
+        0.13774808425903318,
+        0.19241279754638663,
+        1.598234023094177,
+        0.9998
+      ]
+    },
+    "actions": {
+      "mean": [
+        0.0004927651607431471,
+        0.0041174935176968575,
+        0.005378079600632191,
+        -0.0004851108242291957,
+        -0.0010051705176010728,
+        -0.006783927325159311,
+        0.5936462879180908,
+        0.0005593731766566634,
+        -0.00140193453989923,
+        -0.002942675491794944,
+        0.00041861183126457036,
+        0.0004222682327963412,
+        0.0030385160353034735,
+        -0.000329385424265638,
+        -0.0005693831481039524,
+        -0.00029180958517827094,
+        0.000576534541323781,
+        -0.00015302293468266726,
+        -2.4814380594762042e-6,
+        -0.00026009284192696214,
+        0.0003946665092371404,
+        0.000059152920584892854,
+        -0.00006962364568607882,
+        -0.0003498161968309432,
+        0.000053506064432440326,
+        0.00006277022475842386,
+        0.0003592480788938701,
+        -0.00016934213635977358,
+        -0.0001320719748036936,
+        -0.00006360372208291665,
+        0.0001631096820347011,
+        -0.00003750254472834058,
+        -0.000011555765922821593,
+        -0.00006809693149989471,
+        0.00015906375483609736,
+        0.000027621259505394846,
+        -0.00002624911030579824,
+        -0.00012353967758826911,
+        0.000012651072211156134,
+        0.000011669268133118749,
+        0.00012300137314014137,
+        0.00015041981532704085,
+        -0.000056855304137570783,
+        -0.000029746257496299222,
+        0.00007242687570396811,
+        -0.000018986645955010317,
+        -4.556677140499232e-6,
+        -0.00003111416663159616,
+        -0.00009099271846935153,
+        0.000012571626029966865,
+        -0.000013306255823408719,
+        -0.00006410608330043033,
+        6.341869266179856e-6,
+        7.46843352317228e-6,
+        0.00006183652294566855,
+        5.868077096238267e-6,
+        -0.000031927749660098925,
+        -0.00001673644146649167,
+        0.00004087621346116066,
+        -0.000010321637091692537,
+        -2.42824648921669e-6,
+        -0.00001607570084161125,
+        0.00004234428342897445,
+        7.407979410345433e-6,
+        -8.161020559782628e-6,
+        -0.00003870665022986941,
+        3.6879464460071176e-6,
+        4.935619017487625e-6,
+        0.00003818826371571049,
+        0.000051191724196542054,
+        -0.000020595705791492946,
+        -0.000010875675798160955,
+        0.000025950239432859235,
+        -6.903889698151033e-6,
+        -1.2846402341892826e-6,
+        -9.873656381387264e-6,
+        0.00005320658965501934,
+        4.920636911265319e-6,
+        -5.623744073091075e-6,
+        -0.000025824952899711207,
+        2.285133632540237e-6,
+        3.112949343631044e-6,
+        0.00002514909101591911,
+        0.00001734476609271951,
+        -0.000013882496205042116,
+        -7.167188869061647e-6,
+        0.00001790117130440194,
+        -4.3851646296388935e-6,
+        -9.349093375021766e-7,
+        -8.112259820336476e-6,
+        5.336457888915902e-6,
+        3.3939111290237634e-6,
+        -4.019582320324844e-6,
+        -0.00001823278762458358,
+        1.6200752952499897e-6,
+        2.1453445242514135e-6,
+        0.000018565873688203283,
+        0.00004305976108298637,
+        -0.000010276943612552714,
+        -5.3131607273826376e-6,
+        0.000013141003364580683,
+        -3.3397307106497465e-6,
+        -7.699964044149965e-7,
+        -5.316836904967204e-6,
+        -0.00004110661393497139,
+        2.852339093806222e-6,
+        -2.3958800738910213e-6,
+        -0.000013731451872445177,
+        1.5232127452691202e-6,
+        1.386316398566123e-6,
+        0.000013007000234210864,
+        0.00003866312908940017,
+        -7.723581802565604e-6,
+        -3.880812528223032e-6,
+        9.895188668451738e-6,
+        -2.4477972146996763e-6,
+        -5.607461162071559e-7,
+        -3.6636674849432893e-6,
+        0.000010404156455479097,
+        2.0179843431833433e-6,
+        -2.3253292056324426e-6,
+        -0.000010352971003158018,
+        1.0408912203274667e-6,
+        1.1318675205984619e-6,
+        9.978723028325476e-6,
+        -9.606495950720273e-6,
+        -5.949066235189093e-6,
+        -2.8392857984727016e-6,
+        7.4962795224564616e-6,
+        -1.9217382032366004e-6,
+        -4.780694098371896e-7,
+        -3.34773517352005e-6,
+        0.00002107612817781046,
+        1.6454457636427833e-6,
+        -1.817815473259543e-6,
+        -8.123145562421996e-6,
+        7.675727147216094e-7,
+        9.284493671657401e-7,
+        7.868961802159902e-6,
+        -9.203896524923039e-7,
+        -4.736490154755302e-6,
+        -2.4710741399758263e-6,
+        6.142801339592552e-6,
+        -1.6350784335372737e-6,
+        -4.335802543664613e-7,
+        -1.4092432820689282e-6,
+        0.00001656184576859232,
+        1.3327525039130705e-6,
+        -1.3708133792533772e-6,
+        -6.531353847094579e-6,
+        5.90599142924475e-7,
+        6.878227623019484e-7,
+        6.5318749875586946e-6,
+        -4.229010301060043e-6,
+        -4.000812168669654e-6,
+        -2.2653048290521838e-6,
+        5.042587417847244e-6,
+        -1.4821022205069312e-6,
+        -3.451863506143127e-7,
+        -2.389324890827993e-6,
+        8.16222745925188e-6,
+        1.1428018069636892e-6,
+        -1.1332826943544205e-6,
+        -5.504417913471116e-6,
+        5.024350002713618e-7,
+        5.467733217301429e-7,
+        5.279353445075685e-6,
+        -0.0000211214974115137,
+        -3.1678093819209607e-6,
+        -1.8488366322344518e-6,
+        4.2470492189750075e-6,
+        -9.783543646335602e-7,
+        -1.83213785476255e-7,
+        -1.4789864053454949e-6,
+        0.000013087515981169418,
+        7.242887818392774e-7,
+        -1.0046998113466543e-6,
+        -4.340507530287141e-6,
+        3.3989198300332646e-7,
+        5.470611199598352e-7,
+        4.507525773078669e-6,
+        -0.000016270019841613248,
+        -2.829020104400115e-6,
+        -1.296591335631092e-6,
+        3.456672175161657e-6,
+        -9.017520028464787e-7,
+        -1.4440175277741218e-7,
+        -1.4689791214550496e-6,
+        0.000021527563148993067,
+        5.846809472132009e-7,
+        -5.933143256697804e-7,
+        -3.612318550949567e-6,
+        3.5949764765064174e-7,
+        4.328513796281186e-7,
+        3.6886276575387456e-6,
+        -0.000021549723896896467,
+        -2.1144403490325203e-6,
+        -1.4159754755382892e-6,
+        2.68504163614125e-6,
+        -7.679741429456044e-7,
+        -5.776412592695124e-8,
+        -1.192243757941469e-6,
+        0.00003148229370708577,
+        5.478635785038932e-7,
+        -6.972362598389736e-7,
+        -3.211202283637249e-6,
+        2.2451455095051642e-7,
+        3.183014598562295e-7,
+        2.939946398328175e-6,
+        4.565895324049052e-6,
+        -1.7405694734407007e-6,
+        -9.741680742081371e-7,
+        2.2756712496629916e-6,
+        -5.214625389271532e-7,
+        -2.7989250384052866e-7,
+        -8.445824164482474e-7,
+        0.00003105050200247206,
+        5.799903988190636e-7,
+        -7.536694397458632e-7,
+        -2.5344147616124246e-6,
+        2.630972346651106e-7,
+        2.9163163617340615e-7,
+        2.3315690214076312e-6,
+        0.000028919430405949242,
+        -1.4791980902373325e-6,
+        -7.087014637363609e-7,
+        2.1661373921233462e-6,
+        -5.724799052586604e-7,
+        -3.0745661661057966e-7,
+        -5.235916091805848e-7,
+        0.000011437900866440032,
+        3.404710184895521e-7,
+        -4.5958552163938293e-7,
+        -2.1246926280582557e-6,
+        2.5418461291337735e-7,
+        3.871034266467177e-7,
+        1.9572280507418327e-6,
+        0.00003196126999682747,
+        -1.3085544878777e-6,
+        -6.04459614805819e-7,
+        1.575823830535228e-6,
+        -4.1182735799338843e-7,
+        -1.1744142369707333e-7,
+        -1.3322799077286618e-6,
+        2.7163066533830715e-6,
+        3.6782887491426663e-7,
+        -2.6708795530794305e-7,
+        -1.72060231307114e-6,
+        1.8409495794458053e-7,
+        3.7163931665418204e-7,
+        1.8107581354342983e-6,
+        0.000013270215276861563,
+        -1.1510371678014053e-6,
+        -6.630299367316184e-7,
+        1.070535518010729e-6,
+        -3.826233978543314e-7,
+        1.1823525625231923e-7,
+        -1.049707634592778e-6,
+        -2.8137444587628124e-6,
+        2.6974137767865614e-7,
+        -4.0497872078049113e-7,
+        -1.4479496712738182e-6,
+        3.8844518712721765e-8,
+        2.0770157149740953e-8,
+        1.7100152263083146e-6,
+        -0.00002849056909326464,
+        -9.183266342915886e-7,
+        -3.5311214219291287e-7,
+        1.1368990726623451e-6,
+        -3.410139868265105e-7,
+        -2.7979771743957826e-7,
+        -9.036052119881788e-7,
+        -8.269290447060484e-6,
+        3.0624855185124034e-7,
+        -1.1787745535229988e-7,
+        -1.2770473176715313e-6,
+        1.6275393477371836e-7,
+        3.3039643199117563e-7,
+        1.3103665423841449e-6,
+        -0.00001745441295497585,
+        -6.018551061970356e-7,
+        -2.794714930587361e-7,
+        9.792224773264024e-7,
+        -2.040594893060188e-7,
+        -3.4366582468692286e-8,
+        -4.681474479184544e-7,
+        -3.3097076084231958e-6,
+        1.669202376319845e-8,
+        -2.7714148131963157e-7,
+        -9.64512764767278e-7,
+        1.1432685909085194e-7,
+        1.4834664341378812e-7,
+        1.1487189794934238e-6,
+        -0.000010153660696232691,
+        -6.140094228612725e-7,
+        -5.253555173112545e-7,
+        5.710655841539847e-7,
+        -1.404835359153367e-7,
+        4.7323091223461233e-8,
+        -5.46001842849364e-7,
+        -1.2161887070760713e-6,
+        1.2690516371094418e-7,
+        -1.3310798863130913e-7,
+        -7.572793379040377e-7,
+        1.7468086355165724e-7,
+        3.3051062331423964e-8,
+        7.131747565836122e-7,
+        -0.00003459135041339323,
+        -4.08122900807939e-7,
+        -1.8256875478073198e-7,
+        5.756255063715798e-7,
+        -1.6261435575870564e-7,
+        -2.021908862559485e-8,
+        -4.1869603251143417e-7,
+        -1.589809244251228e-6,
+        1.1844038283470582e-7,
+        -7.242162070042468e-8,
+        -4.0011059354583267e-7,
+        -2.4760929306921753e-8,
+        -1.3480507732310798e-7,
+        1.0380339290350094e-6,
+        1.6824413933136384e-6,
+        -4.5796767267347605e-7,
+        -2.0989804738746898e-7,
+        4.763822971654008e-7,
+        -1.7666546625605406e-7,
+        -2.998827852707109e-8,
+        3.5782920804194873e-7,
+        -0.00005497613165061921,
+        -4.8836572119626e-8,
+        -2.8484058134381485e-7,
+        -2.0204961970193835e-7,
+        -1.9280198770843526e-8,
+        8.62186144701127e-9,
+        3.8213460129554733e-7,
+        1.619790509721497e-6,
+        -1.449097339900618e-7,
+        -1.7092831683385157e-7,
+        3.423832595217391e-7,
+        -9.912067611139719e-8,
+        5.862289853553193e-8,
+        -7.740770513464668e-8,
+        -7.812944204488304e-6,
+        1.33551338876714e-7,
+        1.7766551252407226e-7,
+        -2.63828656699161e-8,
+        -5.126612023786947e-9,
+        -4.134697562108158e-8,
+        8.136574791706153e-8,
+        -4.875783361057984e-6
+      ],
+      "std": [
+        0.054471205919981,
+        0.07741593569517136,
+        0.06635452806949615,
+        0.02219589799642563,
+        0.03713744506239891,
+        0.2155904769897461,
+        0.44884487986564636,
+        0.022129571065306664,
+        0.030908629298210144,
+        0.027230359613895416,
+        0.009623671881854534,
+        0.015185607597231865,
+        0.08782459795475006,
+        0.11000800132751465,
+        0.0054918802343308926,
+        0.007706818170845509,
+        0.007092009298503399,
+        0.004339421167969704,
+        0.006056886166334152,
+        0.025186194106936455,
+        0.0549297109246254,
+        0.0023341968189924955,
+        0.003104041563346982,
+        0.0028293700888752937,
+        0.002068394096568227,
+        0.0026343404315412045,
+        0.01149396225810051,
+        0.03640395402908325,
+        0.0011078902753069997,
+        0.0015103736659511924,
+        0.0014142218278720975,
+        0.0010499481577426195,
+        0.0013543356908485293,
+        0.005356063600629568,
+        0.027538813650608063,
+        0.0007573548937216401,
+        0.001025661244057119,
+        0.0009198821499012411,
+        0.0005544671439565718,
+        0.0007547291461378336,
+        0.0035150113981217146,
+        0.02213539369404316,
+        0.000455506524303928,
+        0.0006300596869550645,
+        0.0005910721374675632,
+        0.00034599038190208375,
+        0.0004919870407320559,
+        0.002043767599388957,
+        0.018490547314286232,
+        0.00037283002166077495,
+        0.0005109719350002706,
+        0.00045568859786726534,
+        0.00023278511071112007,
+        0.00032812528661452234,
+        0.0016570856096222997,
+        0.015843955799937248,
+        0.0002481868432369083,
+        0.0003480722662061453,
+        0.000325335975503549,
+        0.00017280328029301018,
+        0.0002512190258130431,
+        0.0010802638716995716,
+        0.01408928632736206,
+        0.00022223185806069523,
+        0.00030612919363193214,
+        0.0002718937466852367,
+        0.00012990496179554611,
+        0.00018422518041916192,
+        0.000973312184214592,
+        0.012508493848145008,
+        0.00015761872055009007,
+        0.00022044251090846956,
+        0.0002061627892544493,
+        0.0001057182380463928,
+        0.00015497820277232677,
+        0.0006764288409613073,
+        0.011296260170638561,
+        0.00014791855937801301,
+        0.00020399357890710235,
+        0.00018064411415252835,
+        0.00008491610060445964,
+        0.0001210872214869596,
+        0.0006432573427446187,
+        0.01023485790938139,
+        0.00010906582610914484,
+        0.00015326884749811143,
+        0.000142823948408477,
+        0.00007245635788422078,
+        0.00010654709330992773,
+        0.00046615453902632,
+        0.009492816403508186,
+        0.00010573185136308894,
+        0.0001460392086300999,
+        0.0001292345259571448,
+        0.00006049493094906211,
+        0.00008650264499010518,
+        0.00045887212036177516,
+        0.008779959753155708,
+        0.00007973915489856154,
+        0.00011203208850929514,
+        0.00010434568685013801,
+        0.00005302290082909167,
+        0.00007767079659970477,
+        0.0003416067629586905,
+        0.00814682524651289,
+        0.00007915469905128703,
+        0.00010886993550229818,
+        0.0000966577063081786,
+        0.000045318782213144004,
+        0.00006455113179981709,
+        0.0003425226896069944,
+        0.007630111649632454,
+        0.0000611644281889312,
+        0.00008592260564910248,
+        0.00007935166649986058,
+        0.000040576029277872294,
+        0.00005935502122156322,
+        0.00026016749325208366,
+        0.007179272826761007,
+        0.00006150962872197852,
+        0.00008518873801222071,
+        0.00007461248605977744,
+        0.0000355022020812612,
+        0.0000502788316225633,
+        0.00026474526384845376,
+        0.006768743507564068,
+        0.000048591213271720335,
+        0.00006827081961091608,
+        0.00006262768874876201,
+        0.000032297353754984215,
+        0.000046926012146286666,
+        0.00020619359565898776,
+        0.006469320971518755,
+        0.00004949716094415635,
+        0.00006947917427169159,
+        0.00005963100556982681,
+        0.000028536098398035392,
+        0.000040751590859144926,
+        0.00021295368787832558,
+        0.006215229630470276,
+        0.00003966964141000062,
+        0.00005751234493800439,
+        0.00005083374344394542,
+        0.00002690069413802121,
+        0.00003854740134556778,
+        0.00016899738693609834,
+        0.005915363319218159,
+        0.00004071096918778494,
+        0.00005744508234784007,
+        0.000049411490181228146,
+        0.000024393977582803927,
+        0.000034359978599241003,
+        0.00017412411398254335,
+        0.005679433699697256,
+        0.000034098236938007176,
+        0.00004947670822730288,
+        0.0000428831554017961,
+        0.000023318862076848745,
+        0.0000333583120664116,
+        0.00014037704386282712,
+        0.005502832122147083,
+        0.000035476441553328186,
+        0.00005029448220739141,
+        0.000041998053347924724,
+        0.000021735353584517725,
+        0.00003051272506127134,
+        0.00014704464410897344,
+        0.005227266810834408,
+        0.000031422849133377895,
+        0.0000454086039098911,
+        0.000037437119317473844,
+        0.000021104986444697715,
+        0.0000290283169306349,
+        0.00012362531560938805,
+        0.005091419909149408,
+        0.00003195118915755302,
+        0.00004717931369668804,
+        0.00003730821481440216,
+        0.000018998216546606272,
+        0.00002651557770150248,
+        0.0001294043322559446,
+        0.004971646703779697,
+        0.00002650190617714543,
+        0.000040163333324017,
+        0.00003248209395678714,
+        0.000018027179976343177,
+        0.00002582307206466794,
+        0.00010801247117342427,
+        0.004754771012812853,
+        0.00002661882717802655,
+        0.00003864263999275863,
+        0.00003127731179120019,
+        0.000016530248103663325,
+        0.000023721948309685104,
+        0.00010995724005624652,
+        0.004578577820211649,
+        0.000022862250261823647,
+        0.000033582466130610555,
+        0.000027823365599033423,
+        0.000015542551409453154,
+        0.000022585352780879475,
+        0.00009167923417408019,
+        0.004508121404796839,
+        0.00002280540320498403,
+        0.000033627373341005296,
+        0.00002707460953388363,
+        0.000014449000445893034,
+        0.00002091616806865204,
+        0.0000951679321588017,
+        0.004384348168969154,
+        0.00002045510154857766,
+        0.00002992084773723036,
+        0.000024755014237598516,
+        0.000014510045730276033,
+        0.000020438790670596063,
+        0.00008195763803087175,
+        0.004295016173273325,
+        0.000021367646695580333,
+        0.00003022674354724586,
+        0.000024624087018310092,
+        0.000013480367670126725,
+        0.000019035556761082262,
+        0.0000849158750497736,
+        0.004229281097650528,
+        0.000018941680536954664,
+        0.000027560265152715147,
+        0.000022634478227701038,
+        0.000013048511391389184,
+        0.00001906735087686684,
+        0.00007548843132099137,
+        0.004119527991861105,
+        0.00002021879299718421,
+        0.00002808637873386033,
+        0.000022752908989787102,
+        0.000012890232028439641,
+        0.000019045859517063946,
+        0.0000775336375227198,
+        0.004028849303722382,
+        0.00001860819975263439,
+        0.000026566647647996433,
+        0.000021319850930012763,
+        0.000012895960026071407,
+        0.00001921564035001211,
+        0.0000719531744834967,
+        0.003980362322181463,
+        0.000019542083464330062,
+        0.000027795384085038677,
+        0.000021858841137145646,
+        0.000013172734725230839,
+        0.000019754204913624562,
+        0.00007459476182702929,
+        0.003902778960764408,
+        0.00002043036147370003,
+        0.00003195027602487244,
+        0.000024560755264246836,
+        0.000014010387531016022,
+        0.00002065290027530864,
+        0.00007676216046093032,
+        0.0038293309044092894,
+        0.000023211092411656864,
+        0.00003743018169188872,
+        0.000025842035029199906,
+        0.000015010549759608693,
+        0.000020894838598906063,
+        0.00008554229862056673,
+        0.0037848083302378654,
+        0.00002203849180659745,
+        0.00003869964712066576,
+        0.00002533109909563791,
+        0.000013811703865940217,
+        0.000020183759261271916,
+        0.0000796892709331587,
+        0.0037312512286007404,
+        0.000020139470507274382,
+        0.00003185970854246989,
+        0.00002326746835024096,
+        0.000014411976735573262,
+        0.000020371249775053002,
+        0.00007360208837781101,
+        0.0036642765626311302,
+        0.000017049775124178268,
+        0.000027688640329870395,
+        0.00001989392148971092,
+        0.000012698577847913839,
+        0.000017972832210944034,
+        0.00006560993642779067,
+        0.0036550285294651985,
+        0.000016028958270908333,
+        0.000024884970116545446,
+        0.00001818954842747189,
+        0.000011278674719505943,
+        0.000015812189303687774,
+        0.00006195339665282518,
+        0.0036316923797130585,
+        0.00001560178498039022,
+        0.000024082159143290482,
+        0.00001761745625117328,
+        0.00001089257057174109,
+        0.000015621577404090203,
+        0.00005938451431575231,
+        0.003624820150434971,
+        0.000014879788068355992,
+        0.00002372561357333325,
+        0.00001737473940011114,
+        0.00001018085913528921,
+        0.000015368110325653106,
+        0.00005791655712528154,
+        0.0035625966265797615,
+        0.000014492210539174266,
+        0.000023905164198367856,
+        0.000016788300854386762,
+        0.00001016644364426611,
+        0.000014378600099007599,
+        0.00005657917427015491,
+        0.0035458113998174667,
+        0.000014646487215941306,
+        0.000023379501726594754,
+        0.000016420719475718215,
+        0.000010098652637680061,
+        0.000014440963241213467,
+        0.00005597082417807542,
+        0.0035429371055215597,
+        0.000014022874893271364,
+        0.00002083050821966026,
+        0.000015985659047146328,
+        0.00001016553051158553,
+        0.000013916131138103083,
+        0.000056019474868662655,
+        0.0034998103510588408,
+        0.000014245401871448848,
+        0.000022099453417467885,
+        0.00001657450047787279,
+        0.000010621713045111392,
+        0.000014651923265773803,
+        0.00005747186514781788,
+        0.0034743566066026688,
+        0.000014478311641141772,
+        0.00002349081842112355,
+        0.000017406200640834868,
+        0.000010810344974743202,
+        0.000014864897821098566,
+        0.00005688491728506051,
+        0.0034770709462463856,
+        0.000016515868992428295,
+        0.00002587365270301234,
+        0.000018931375961983576,
+        0.000010814174856932368,
+        0.000016132373275468126,
+        0.00006046127964509651,
+        0.003478288883343339
+      ],
+      "q01": [
+        -0.15909142580032348,
+        -0.23591104531884194,
+        -0.1173719973474741,
+        -0.0725411132991314,
+        -0.12759929532110692,
+        -0.6792203711509704,
+        0.0,
+        -0.06588815684318543,
+        -0.1043403072744608,
+        -0.07569241938591004,
+        -0.026123490297794344,
+        -0.04355562012195587,
+        -0.26860033273696904,
+        -0.31594268302917483,
+        -0.017998227071762084,
+        -0.025448361220955847,
+        -0.021794256627559662,
+        -0.014516740383952856,
+        -0.017180888187885286,
+        -0.08126688668131828,
+        -0.15649057692885399,
+        -0.006211889632977545,
+        -0.009136751361563802,
+        -0.00862356810811907,
+        -0.00636542599722743,
+        -0.007391496415808798,
+        -0.035477512252330784,
+        -0.10543232785761356,
+        -0.003451917093619704,
+        -0.004779039949178695,
+        -0.00400189593732357,
+        -0.003541165339201688,
+        -0.003899161383137107,
+        -0.016611159443855286,
+        -0.07849292940199376,
+        -0.001959954298567027,
+        -0.0028813693393021818,
+        -0.0027899963006842885,
+        -0.0015735435530543325,
+        -0.002058688412979245,
+        -0.010417635813355447,
+        -0.0639662309139967,
+        -0.0013655057244002819,
+        -0.0019750579357147215,
+        -0.0016051418443210422,
+        -0.0011298993272706869,
+        -0.0014582404736429451,
+        -0.006558569058775902,
+        -0.05251251416504383,
+        -0.0009643727889051661,
+        -0.0014315907768905163,
+        -0.001390606227517128,
+        -0.0006525630556046963,
+        -0.0008888287223875523,
+        -0.004895182948186994,
+        -0.0455008171916008,
+        -0.0007486765272915363,
+        -0.0010981693975627422,
+        -0.000882065705396235,
+        -0.0005725251567782834,
+        -0.0007578355811536312,
+        -0.0034509299120865766,
+        -0.04015725544691086,
+        -0.0005749710650416091,
+        -0.0008555282396264373,
+        -0.0008345671106129884,
+        -0.0003548953181132674,
+        -0.0004969418092630804,
+        -0.0028254041438922285,
+        -0.035569155138731004,
+        -0.0004705392210977152,
+        -0.0006840464788489043,
+        -0.0005612756710499526,
+        -0.00035702317333780227,
+        -0.0004648537166416645,
+        -0.002128881996870041,
+        -0.03083838468044997,
+        -0.00038048800966935235,
+        -0.000566111233481206,
+        -0.0005538229705183766,
+        -0.00023378437627106905,
+        -0.00032636490603908896,
+        -0.0018706658044829966,
+        -0.02931403119564057,
+        -0.00032479510342236606,
+        -0.000475912622641772,
+        -0.00038455819835653526,
+        -0.00024285348085686562,
+        -0.0003135723091545515,
+        -0.001452100250311196,
+        -0.026792802512645722,
+        -0.00027214939426630733,
+        -0.00040756029058247796,
+        -0.0003960524151916616,
+        -0.00016536210232879968,
+        -0.00023339646961539982,
+        -0.0013170160830486568,
+        -0.02499642580747604,
+        -0.00024108570310054343,
+        -0.000348888192162849,
+        -0.0002832006874959916,
+        -0.00017626483961939813,
+        -0.00023037131156306717,
+        -0.0010707858602516352,
+        -0.023444418999552727,
+        -0.00020677300081588327,
+        -0.0003006568572251126,
+        -0.0002946901318966411,
+        -0.00012258605094393713,
+        -0.00017606193688698117,
+        -0.0009890523520298302,
+        -0.02203794540464878,
+        -0.0001841578755469527,
+        -0.00026803012369200585,
+        -0.00021574946893379094,
+        -0.00013577964630676436,
+        -0.0001734850220615044,
+        -0.0008068450814578683,
+        -0.020441363253444433,
+        -0.00016045538827893325,
+        -0.000241462552268058,
+        -0.00022808200246654452,
+        -0.00009496491550235078,
+        -0.00013633702206425368,
+        -0.0007666170152835548,
+        -0.019491385824978353,
+        -0.0001477667265571654,
+        -0.00021141915648477145,
+        -0.0001735861329594627,
+        -0.00010635797707363962,
+        -0.00013736910806037486,
+        -0.0006427704328671098,
+        -0.018371846283972263,
+        -0.0001293026403174736,
+        -0.00019473347970051693,
+        -0.0001822685490187723,
+        -0.00007863951541366989,
+        -0.00010972291373182087,
+        -0.0006055113250389692,
+        -0.017673968444019555,
+        -0.00011848664400167762,
+        -0.00017848168123746295,
+        -0.00014009435730986296,
+        -0.0000859230127884075,
+        -0.00011260414165444673,
+        -0.0005137789773289115,
+        -0.016190811094641686,
+        -0.00010690572904422879,
+        -0.0001638943360070698,
+        -0.0001501999743981287,
+        -0.00006396331703872426,
+        -0.00009358972666086629,
+        -0.0005007586104329676,
+        -0.01619275164604187,
+        -0.00010315810496686024,
+        -0.00015018478848505767,
+        -0.00011576190601917915,
+        -0.00007506265956908466,
+        -0.00009504533028230073,
+        -0.0004348065205849708,
+        -0.01565815359354019,
+        -0.00009185501914471398,
+        -0.0001392430538544431,
+        -0.00012555517936125394,
+        -0.00005638334401883189,
+        -0.00008002441792050375,
+        -0.000424562167050317,
+        -0.015019814673811197,
+        -0.00009095547425094991,
+        -0.00013837829248514026,
+        -0.00010207304285722785,
+        -0.00006334332783007992,
+        -0.00008174893613904723,
+        -0.0003743791728978976,
+        -0.014377124969661235,
+        -0.00008569652531296014,
+        -0.00013066508993506436,
+        -0.00010974786025471983,
+        -0.000049689067096915106,
+        -0.00007135935623664408,
+        -0.0003742531020194291,
+        -0.014187388128042218,
+        -0.0000794089021859691,
+        -0.00011592403496615584,
+        -0.0000893614823697135,
+        -0.0000560631047235802,
+        -0.0000719853327609599,
+        -0.0003315736224874853,
+        -0.013546264378726483,
+        -0.0000717077999375761,
+        -0.00010777361229993398,
+        -0.0000922710225568153,
+        -0.00004497638295288198,
+        -0.00006191108026541768,
+        -0.0003081054835114628,
+        -0.013243352345749734,
+        -0.00006864526765421036,
+        -0.0001027144630090333,
+        -0.00008054810634930621,
+        -0.000049568866193294524,
+        -0.00006090367061551658,
+        -0.00028183929068036377,
+        -0.012778499507904054,
+        -0.00006117501191329209,
+        -0.00009796037861378861,
+        -0.00008128996908781118,
+        -0.00004006202814634888,
+        -0.00005725241685286168,
+        -0.0002700211696093902,
+        -0.012565325374901296,
+        -0.00006034678600262851,
+        -0.0000907877254066989,
+        -0.00007010409524082204,
+        -0.000041811013757251194,
+        -0.0000556492382194847,
+        -0.0002493678867351264,
+        -0.011774876400828364,
+        -0.00005579026520717887,
+        -0.00008717787461355332,
+        -0.00007443824684014546,
+        -0.0000356287220492959,
+        -0.00005372272633831019,
+        -0.00024090840241406124,
+        -0.011997990143299102,
+        -0.000058186027396004656,
+        -0.00008279621332185345,
+        -0.00006459413380362097,
+        -0.00003831381897907702,
+        -0.00005259611281217076,
+        -0.00022310287454165522,
+        -0.011667696542292834,
+        -0.00005284995273686948,
+        -0.00008384203007444737,
+        -0.00006744749160716312,
+        -0.000032252046390203774,
+        -0.0000481185418262612,
+        -0.00021629847201984375,
+        -0.011543062198162084,
+        -0.00005847320245811716,
+        -0.00008175599463284015,
+        -0.00006548594534397124,
+        -0.000038102848210837704,
+        -0.0000492544468608685,
+        -0.00021983185317367318,
+        -0.011402166564017536,
+        -0.00005387163197156041,
+        -0.0000825874566333369,
+        -0.00006687336476170457,
+        -0.000033979644608916735,
+        -0.00005101760845864191,
+        -0.00020870170197449626,
+        -0.011224610126763582,
+        -0.00006089058108045721,
+        -0.00008930324914399536,
+        -0.00007119371397420766,
+        -0.00003768286973936482,
+        -0.000053939796844497314,
+        -0.0002299774368293584,
+        -0.010891138190776108,
+        -0.000060360829543788016,
+        -0.00009397184122353786,
+        -0.00007481435872614382,
+        -0.00003693910902366043,
+        -0.00005306754379416818,
+        -0.0002138554411940277,
+        -0.01082267729640007,
+        -0.00005871810513781385,
+        -0.00008991035101935281,
+        -0.00007033640673034824,
+        -0.000036662435624748536,
+        -0.00005430486062541601,
+        -0.0002200629871338608,
+        -0.010579705061763527,
+        -0.000053560086898505666,
+        -0.00008582322522997861,
+        -0.00007200759416446087,
+        -0.000034787034674081956,
+        -0.000048553584120236346,
+        -0.0002130190141499044,
+        -0.010566252829506993,
+        -0.00005062428400851786,
+        -0.00007914942941861225,
+        -0.000059331482945708576,
+        -0.00003430225813062863,
+        -0.00004731270743068308,
+        -0.00019290218278765676,
+        -0.010022874112427235,
+        -0.000047412463498767456,
+        -0.00007506768405437467,
+        -0.00005783305459190161,
+        -0.00002997652450576426,
+        -0.00004301143199554646,
+        -0.00018155146474018715,
+        -0.010345511507242918,
+        -0.00004604501994326713,
+        -0.00007326857978478065,
+        -0.00005605542755220084,
+        -0.000028739307308569553,
+        -0.00004117103526368737,
+        -0.0001768201191909612,
+        -0.010309934300184249,
+        -0.00004275882025249304,
+        -0.00006731156632304187,
+        -0.00005409912285394967,
+        -0.000026329884584993103,
+        -0.00004337953124195339,
+        -0.00016598801976069823,
+        -0.01018075596317649,
+        -0.000043296541273593885,
+        -0.00006625080597586939,
+        -0.00005305007540155199,
+        -0.000028063857986126094,
+        -0.000040029496920760744,
+        -0.00017517980523407465,
+        -0.010162209048867225,
+        -0.0000426195073465351,
+        -0.00006396621902240442,
+        -0.000051819634152343525,
+        -0.000029033373587299122,
+        -0.00004320319109829139,
+        -0.00016554977782070636,
+        -0.010125353442132473,
+        -0.000045235728612169645,
+        -0.00006515037268400192,
+        -0.00004881905938964337,
+        -0.000028501104679889977,
+        -0.00004041612264700232,
+        -0.00016903663412667815,
+        -0.010062240642309189,
+        -0.000043975374847650503,
+        -0.00006714530037716033,
+        -0.00005093284538015727,
+        -0.00002801080799545163,
+        -0.00004157804111018775,
+        -0.00016746679800562535,
+        -0.009975232541561126,
+        -0.000045126923860516424,
+        -0.00007225614520721142,
+        -0.00005288218818604951,
+        -0.000029170855134725583,
+        -0.000040984369814395916,
+        -0.0001663898449391129,
+        -0.009851353891193867,
+        -0.00004605035483837126,
+        -0.00007112415134906766,
+        -0.0000582737371325493,
+        -0.000028596166521310774,
+        -0.000044461411982774806,
+        -0.0001793744623661041,
+        -0.009932486386597156
+      ],
+      "q99": [
+        0.16212226042747496,
+        0.25229084351658815,
+        0.18486946655809877,
+        0.061344154804944984,
+        0.10886132830679413,
+        0.6657196632862092,
+        0.9998,
+        0.06576045138835906,
+        0.09603128664791583,
+        0.045872775423526774,
+        0.03236173896193503,
+        0.05487722767591477,
+        0.2764705369472503,
+        0.3158153381347656,
+        0.014891946853697303,
+        0.023800758841633796,
+        0.018585946798324586,
+        0.012693745650351043,
+        0.021670551879704,
+        0.08171248322725297,
+        0.15637542012631894,
+        0.0069066042726859435,
+        0.009335029612295327,
+        0.0047039436949416966,
+        0.006996506480127576,
+        0.008702572771534321,
+        0.036967494320869446,
+        0.1053792189329863,
+        0.002857540221139789,
+        0.004296822899580001,
+        0.003630450583249331,
+        0.003087930056080223,
+        0.004339818637073041,
+        0.01760330287218094,
+        0.07830542102754112,
+        0.0021181412500329315,
+        0.0029414672447368503,
+        0.0015112007031682874,
+        0.0019269651249051098,
+        0.0024494722973555337,
+        0.01144827667623758,
+        0.06380807581543924,
+        0.0011728783883154393,
+        0.001813931218627841,
+        0.0014501392262987797,
+        0.0009983487627469,
+        0.0015205995496362446,
+        0.006439514055848122,
+        0.05315425983667374,
+        0.0010328240715665742,
+        0.0014663059995509691,
+        0.0007404281431809068,
+        0.0008001560054719447,
+        0.001095746040716767,
+        0.0052222565246745935,
+        0.04538371631056071,
+        0.0006383903683163227,
+        0.0009848102391697465,
+        0.000784420038573444,
+        0.0004887675133766608,
+        0.0007893593505024911,
+        0.0033699059166945513,
+        0.039476715770363824,
+        0.0006163579971762373,
+        0.0008800703473389147,
+        0.00044086696542799475,
+        0.000447366009419784,
+        0.0006247742644743994,
+        0.0030928143300116076,
+        0.035530846929550164,
+        0.0004072732983855529,
+        0.0006279987238347533,
+        0.0004935565162450075,
+        0.0003003585597965866,
+        0.0004761313281953334,
+        0.002095991748571396,
+        0.030764527727663513,
+        0.00040901676943758494,
+        0.0005783375784289091,
+        0.00029091126989806075,
+        0.0002924530396237969,
+        0.0004053453548112883,
+        0.0020267377858981485,
+        0.02924695993065833,
+        0.0002835550721851177,
+        0.0004330007825046778,
+        0.0003438553759129719,
+        0.00021280083870515228,
+        0.0003332204633043148,
+        0.0014287737663835282,
+        0.02666576117277146,
+        0.00029395890273153794,
+        0.00041647118404507634,
+        0.0002144097874115686,
+        0.00020652228510007256,
+        0.0002845425325096586,
+        0.0014505576011724767,
+        0.024956510066986085,
+        0.00020434754303423695,
+        0.00031225532339885847,
+        0.0002542992958333343,
+        0.0001512215134222061,
+        0.00024444468863075595,
+        0.0010491569862701002,
+        0.023069663181155914,
+        0.00022068456518463788,
+        0.00031388382902368915,
+        0.00016011388470651587,
+        0.00015496468472993004,
+        0.00021093179946765303,
+        0.001086882085958496,
+        0.0219863745495677,
+        0.00016183016332215621,
+        0.00024126361515372995,
+        0.0001919538950081915,
+        0.00011462044673971831,
+        0.00018409015340730545,
+        0.0008034803080838172,
+        0.020384925924241543,
+        0.0001725941591721494,
+        0.00024417606623610485,
+        0.00012369809069205076,
+        0.00012043028190964833,
+        0.00016321412267861885,
+        0.0008335524674970658,
+        0.019483701366186136,
+        0.00012494574459269637,
+        0.0001922543402411975,
+        0.00015176694963593034,
+        0.00008988425261923113,
+        0.00014705147650092838,
+        0.0006307619461789725,
+        0.01831897035390139,
+        0.00013962995521724217,
+        0.00019706185359973463,
+        0.00010126548151019962,
+        0.0000950566966785118,
+        0.00013232208000263205,
+        0.000660218992456794,
+        0.01764387315139175,
+        0.00010247177979908883,
+        0.00015943087645573536,
+        0.00012344940464245158,
+        0.00007527396054356363,
+        0.00011956113649066536,
+        0.0005160252890083939,
+        0.016168088349699967,
+        0.00011438966927817092,
+        0.00016185711201978847,
+        0.00008490921122720466,
+        0.00007747934755752788,
+        0.00010678794741397725,
+        0.0005481199477799237,
+        0.01615487493276596,
+        0.00008468688050634237,
+        0.00013357372635509814,
+        0.00010555801286245698,
+        0.00006330196186900142,
+        0.000101998839713633,
+        0.00042467831117101035,
+        0.015395162180066112,
+        0.00009998356355354183,
+        0.00014026010758243506,
+        0.00007274458486353978,
+        0.00006777426579501484,
+        0.00009303196137771012,
+        0.0004640237890882414,
+        0.01499328883215785,
+        0.00007716326334048071,
+        0.00011421011368511244,
+        0.00009314750883495434,
+        0.00005738810400944202,
+        0.00008783590141683817,
+        0.0003717806453816595,
+        0.014554040935635573,
+        0.00008682058225385845,
+        0.00012317761406302453,
+        0.00007470068163238468,
+        0.00005833794826176018,
+        0.0000807305663009174,
+        0.0003966953692026436,
+        0.014136480166763069,
+        0.00006818514921469611,
+        0.00010206100093200813,
+        0.00008256861115805807,
+        0.000049349277792498495,
+        0.00007737125915009528,
+        0.00031947106840088976,
+        0.013474849896132943,
+        0.00007525306886527682,
+        0.00010829368010163308,
+        0.00006185477034305223,
+        0.000050571078446228043,
+        0.00007023279787972571,
+        0.0003371094720438124,
+        0.013198494754731657,
+        0.000059138852264732124,
+        0.00008794275637483227,
+        0.00007062362715369087,
+        0.00004125927500426771,
+        0.00006884480739245192,
+        0.0002681864194571972,
+        0.012739881089329719,
+        0.00006564935145433995,
+        0.00009313276901375502,
+        0.00005367935128742829,
+        0.00004326853083912283,
+        0.000059486858663149135,
+        0.000290332669764757,
+        0.012545088732242582,
+        0.00005332418816396964,
+        0.0000798318615881726,
+        0.00006209178492426877,
+        0.00003812435691943395,
+        0.00005920125464908778,
+        0.00024440587060526003,
+        0.011752966091036793,
+        0.00006044615641003474,
+        0.00008350273473188281,
+        0.00005087533958721905,
+        0.00004026508983224629,
+        0.00005588843687437471,
+        0.0002512834585038945,
+        0.01198878407478332,
+        0.00004965451551834123,
+        0.00007573206002125512,
+        0.00006231120838783678,
+        0.00003364615770988169,
+        0.00005282912561669945,
+        0.0002209204529877752,
+        0.011624232469499118,
+        0.00005783643363974989,
+        0.00008142779953777802,
+        0.00005095888577052386,
+        0.000036499108670977874,
+        0.000052707964507862936,
+        0.000231120909564197,
+        0.01148185554742813,
+        0.00004983878775965424,
+        0.00007604824067093425,
+        0.000057698829320725065,
+        0.000033903982862830136,
+        0.000054097583203110825,
+        0.00020294103957712662,
+        0.01119744649901986,
+        0.00005782577866921202,
+        0.00007854039029916743,
+        0.0000542719262011815,
+        0.00003704023372847582,
+        0.00005206325108883899,
+        0.00022946770491544147,
+        0.011211802294850345,
+        0.0000541387221601326,
+        0.00008076134007424106,
+        0.00006221706023206938,
+        0.00003468251589220018,
+        0.000056180346361361434,
+        0.00020245690247975277,
+        0.010855805838853122,
+        0.00005655899195699014,
+        0.0000880048372084274,
+        0.00006508456927258521,
+        0.00003773696059361102,
+        0.00005296763135702342,
+        0.00022277740947902205,
+        0.010783490767329935,
+        0.00005639911381294957,
+        0.00008836033511906856,
+        0.00007074359970283693,
+        0.00003449545800685873,
+        0.00004960680939257152,
+        0.0002082344447262587,
+        0.010557891380786895,
+        0.00005552569483406838,
+        0.00008377498388290397,
+        0.0000605235195718706,
+        0.0000366905065602623,
+        0.00005407085546758027,
+        0.00021638836078345788,
+        0.010525680954381826,
+        0.00004830919423839074,
+        0.00007733448530780152,
+        0.00005978569525177588,
+        0.00003123643363360317,
+        0.00004709080595057457,
+        0.0001939378370530904,
+        0.00998856540247798,
+        0.00004645563278463669,
+        0.00007126600248739121,
+        0.00005200477748876437,
+        0.00003055265350267292,
+        0.000045497189089655854,
+        0.00018253272622823706,
+        0.01032583272829652,
+        0.000044604689953848674,
+        0.00006673079421743756,
+        0.000050660950923338505,
+        0.000027101854892680433,
+        0.000045762255322188115,
+        0.00016890384713187808,
+        0.010136223459243775,
+        0.00004455379862338299,
+        0.00006405452857725328,
+        0.00004943669138010592,
+        0.000028590407309820904,
+        0.000039900707779452174,
+        0.00017483096104115243,
+        0.010142471250146626,
+        0.0000416521602543071,
+        0.00006346322163008151,
+        0.00005030547470669263,
+        0.00002662300992524252,
+        0.00003956695257220416,
+        0.00016375841144472352,
+        0.009997803356498476,
+        0.000042872124351561063,
+        0.00006597544132964689,
+        0.00005052208236302249,
+        0.00002720812879269946,
+        0.000039130584619124446,
+        0.0001802312535233797,
+        0.010092107820510864,
+        0.00003863017375115306,
+        0.00006420832830481232,
+        0.000052780072088353346,
+        0.000026064168196171522,
+        0.000040864129550755046,
+        0.00018467701834160846,
+        0.009899575112760066,
+        0.0000431866092141718,
+        0.00006625963510014114,
+        0.00005088201165199277,
+        0.000026399783365195603,
+        0.000037794473674148364,
+        0.00017790157631970968,
+        0.00994323554635048,
+        0.00004512640612665562,
+        0.00007162263081409033,
+        0.00005516039710491889,
+        0.000026367472112178792,
+        0.00004286482669413089,
+        0.0001678395111113784,
+        0.009839660620689389,
+        0.000047105535864830025,
+        0.00007688079774379733,
+        0.00005521634444594385,
+        0.00002967622131109241,
+        0.000041797038167715075,
+        0.00017254053354263292,
+        0.009918793506920338
+      ]
+    }
+  },
+  "sample_ratio": 0.05,
+  "dataset_infos": {
+    "robotgeneralist/pp__1": {
+      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
+      "num_episodes": 237,
+      "num_frames": 99363
+    },
+    "robotgeneralist/pp__1__adversarial": {
+      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
+      "num_episodes": 64,
+      "num_frames": 17906
+    },
+    "robotgeneralist/pp__2": {
+      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
+      "num_episodes": 92,
+      "num_frames": 37458
+    },
+    "robotgeneralist/pp__2__rotations": {
+      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
+      "num_episodes": 102,
+      "num_frames": 13935
+    },
+    "robotgeneralist/pp__3__with_ft": {
+      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
+      "num_episodes": 99,
+      "num_frames": 38424
+    },
+    "robotgeneralist/pp__3__with_ft__rotations": {
+      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
+      "num_episodes": 49,
+      "num_frames": 7151
+    },
+    "robotgeneralist/pp__3__with_ft__adversarial_start": {
+      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
+      "num_episodes": 147,
+      "num_frames": 21313
+    },
+    "robotgeneralist/pp__4__with_ft": {
+      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
+      "num_episodes": 94,
+      "num_frames": 33046
+    },
+    "robotgeneralist/pp__5__with_ft__adversarial_target": {
+      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
+      "num_episodes": 48,
+      "num_frames": 10615
+    },
+    "robotgeneralist/pp__6__with_ft": {
+      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
+      "num_episodes": 379,
+      "num_frames": 123611
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-1": {
+      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
+      "num_episodes": 98,
+      "num_frames": 41410
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-2": {
+      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
+      "num_episodes": 94,
+      "num_frames": 37859
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-3": {
+      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
+      "num_episodes": 127,
+      "num_frames": 53341
+    },
+    "robotgeneralist/precise-packing-2025-10-22-part-4": {
+      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
+      "num_episodes": 102,
+      "num_frames": 38263
+    }
+  }
+}

pp__ee_fourier_k50/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}

pp__ee_fourier_k50/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}]]}}"}

pp__ee_fourier_k50/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}}]}

pp__ee_fourier_k50/config.yaml

@@ -0,0 +1,108 @@
+!!python/object:openpi.training.config.TrainConfig
+name: precise_packing_FFM_k50
+project_name: openpi
+exp_name: pp__ee_fourier_k50
+model: !!python/object:openpi.models.pi0_config.Pi0Config
+    action_dim: 32
+    action_horizon: 50
+    max_token_len: 48
+    dtype: bfloat16
+    paligemma_variant: gemma_2b
+    action_expert_variant: gemma_300m
+    img_variant: So400m/14
+    action_loss_horizon: 50
+    pi05: false
+    discrete_state_input: false
+weight_loader: !!python/object:openpi.training.weight_loaders.CheckpointWeightLoader
+    local_dir: null
+    repo_id: robotgeneralist/openpi_checkpoint_mirrors
+    revision: main
+    checkpoint_path: pi0_base
+pytorch_weight_path: null
+pytorch_training_precision: bfloat16
+lr_schedule: !!python/object:openpi.training.optimizer.CosineDecaySchedule
+    warmup_steps: 1000
+    peak_lr: 2.5e-05
+    decay_steps: 30000
+    decay_lr: 2.5e-06
+optimizer: !!python/object:openpi.training.optimizer.AdamW
+    b1: 0.9
+    b2: 0.95
+    eps: 1.0e-08
+    weight_decay: 1.0e-10
+    clip_gradient_norm: 1.0
+ema_decay: 0.99
+freeze_filter: !!python/name:flax.nnx.filterlib.Nothing ''
+data: !!python/object:openpi.training.config.LeRobotNomagicURXDataConfig
+    repo_ids: !!python/tuple
+    - robotgeneralist/pp__1
+    - robotgeneralist/pp__1__adversarial
+    - robotgeneralist/pp__2
+    - robotgeneralist/pp__2__rotations
+    - robotgeneralist/pp__3__with_ft
+    - robotgeneralist/pp__3__with_ft__rotations
+    - robotgeneralist/pp__3__with_ft__adversarial_start
+    - robotgeneralist/pp__4__with_ft
+    - robotgeneralist/pp__5__with_ft__adversarial_target
+    - robotgeneralist/pp__6__with_ft
+    - robotgeneralist/precise-packing-2025-10-22-part-1
+    - robotgeneralist/precise-packing-2025-10-22-part-2
+    - robotgeneralist/precise-packing-2025-10-22-part-3
+    - robotgeneralist/precise-packing-2025-10-22-part-4
+    assets: !!python/object:openpi.training.config.AssetsConfig
+        assets_dir: null
+        asset_id: ur10e
+    norm_stats_sample_ratio: 0.05
+    base_config: !!python/object:openpi.training.config.DataConfig
+        repo_ids: null
+        asset_id: null
+        norm_stats_data: null
+        norm_stats_sample_ratio: 1.0
+        norm_stats_dir: null
+        tolerance_s: 0.0001
+        per_action_index_norm_stats: false
+        repack_transforms: !!python/object:openpi.transforms.Group
+            inputs: !!python/tuple []
+            outputs: !!python/tuple []
+        data_transforms: !!python/object:openpi.transforms.Group
+            inputs: !!python/tuple []
+            outputs: !!python/tuple []
+        model_transforms: !!python/object:openpi.transforms.Group
+            inputs: !!python/tuple []
+            outputs: !!python/tuple []
+        use_quantile_norm: false
+        action_sequence_keys: !!python/tuple
+        - actions
+        fourier_config: null
+        prompt_from_task: true
+        rlds_data_dir: null
+        action_space: null
+        filter_dict_path: null
+    default_prompt: null
+    use_poses: true
+    end_effector_frame: true
+    per_action_index_norm_stats: true
+    fourier_config: !!python/object:openpi.fourier.FourierConfig
+        binary_action_keys: !!python/tuple
+        - tool
+        action_horizon: 50
+        num_modes: 50
+        norm: forward
+    tolerance_s: 0.0001
+assets_base_dir: ./assets
+checkpoint_base_dir: /storage_nvme_1/robotgeneralist/checkpoints
+seed: 42
+batch_size: 32
+num_workers: 2
+num_train_steps: 25001
+log_interval: 100
+save_interval: 1000
+keep_period: 5000
+overwrite: false
+resume: true
+wandb_enabled: true
+policy_metadata: null
+fsdp_devices: 1
+test_split_ratio: 0.05
+test_eval_interval: 1000
+test_eval_use_train_mode: false

pp__ee_fourier_k50/wandb_id.txt

@@ -0,0 +1 @@
+8ey01m45