50000/default/_METADATA

{"tree_metadata": {"('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoder_norm', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoder_norm', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_0', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_0", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_1', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_1", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_10', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_10", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_11', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_11", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_2', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_2", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_3', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_3", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_4', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_4", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_5', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_5", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_6', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_6", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_7', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_7", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_8', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_8", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'BlockTransformer_0', 'Transformer_0', 'encoderblock_9', 'MultiHeadDotProductAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "BlockTransformer_0", "key_type": 2}, {"key": "Transformer_0", "key_type": 2}, {"key": "encoderblock_9", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_left_pos_embedding')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_left_pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_left_projection', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_left_projection", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_left_projection', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_left_projection", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_primary_pos_embedding')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_primary_pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_primary_projection', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_primary_projection", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_primary_projection', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_primary_projection", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_single_pos_embedding')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_single_pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_single_projection', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_single_projection", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'obs_single_projection', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "obs_single_projection", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block1', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block2', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block3', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'block4', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'conv_root', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "conv_root", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'gn_root', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "gn_root", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_left', 'ResNet26FILM_0', 'gn_root', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_left", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "gn_root", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_1', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_1", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_0', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'FilmConditioning_2', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "FilmConditioning_2", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block1', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block2', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block2", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block3', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block3", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'conv_proj', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "conv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn_proj', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit1', 'gn_proj', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit1", "key_type": 2}, {"key": "gn_proj", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'conv1', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'conv2', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'conv3', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "conv3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'gn1', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'gn1', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'gn2', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'gn2', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'gn3', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'block4', 'unit2', 'gn3', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "block4", "key_type": 2}, {"key": "unit2", "key_type": 2}, {"key": "gn3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'conv_root', 'kernel')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "conv_root", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'gn_root', 'bias')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "gn_root", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'observation_tokenizers_primary', 'ResNet26FILM_0', 'gn_root', 'scale')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "observation_tokenizers_primary", "key_type": 2}, {"key": "ResNet26FILM_0", "key_type": 2}, {"key": "gn_root", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'readout_bimanual_pos_embedding')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "readout_bimanual_pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'readout_nav_pos_embedding')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "readout_nav_pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'readout_quadruped_pos_embedding')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "readout_quadruped_pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('crossformer_transformer', 'readout_single_arm_pos_embedding')": {"key_metadata": [{"key": "crossformer_transformer", "key_type": 2}, {"key": "readout_single_arm_pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_bimanual', 'mean_proj', 'bias')": {"key_metadata": [{"key": "heads_bimanual", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_bimanual', 'mean_proj', 'kernel')": {"key_metadata": [{"key": "heads_bimanual", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_nav', 'mean_proj', 'bias')": {"key_metadata": [{"key": "heads_nav", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_nav', 'mean_proj', 'kernel')": {"key_metadata": [{"key": "heads_nav", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_quadruped', 'mean_proj', 'bias')": {"key_metadata": [{"key": "heads_quadruped", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_quadruped', 'mean_proj', 'kernel')": {"key_metadata": [{"key": "heads_quadruped", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_single_arm', 'mean_proj', 'bias')": {"key_metadata": [{"key": "heads_single_arm", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}, "('heads_single_arm', 'mean_proj', 'kernel')": {"key_metadata": [{"key": "heads_single_arm", "key_type": 2}, {"key": "mean_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": true}}}}