{"name": "model_1", "trainable": true, "layers": [{"module": "keras.layers", "class_name": "InputLayer", "config": {"batch_input_shape": [null, 8192], "dtype": "int32", "sparse": false, "ragged": false, "name": "input_2"}, "registered_name": null, "name": "input_2", "inbound_nodes": []}, {"module": "tokun.model", "class_name": "Encoder", "config": {"token_dim": [4, 4], "encoding_dim": 256, "embedding_dim": 256, "sequence_axis": 1, "feature_axis": -1, "activation": "gelu"}, "registered_name": "models>Encoder", "build_config": {"input_shape": [16, 2048]}, "name": "encoder_1", "inbound_nodes": [[["input_2", 0, 0, {}]]]}, {"module": "tokun.model", "class_name": "Decoder", "config": {"token_dim": [4, 4], "encoding_dim": 256, "embedding_dim": 256, "sequence_axis": 1, "feature_axis": -1, "activation": "gelu", "output": "categorical"}, "registered_name": "models>Decoder", "build_config": {"input_shape": [16, 128, 256]}, "name": "decoder_1", "inbound_nodes": [[["encoder_1", 0, 0, {}]]]}], "input_layers": [["input_2", 0, 0]], "output_layers": [["decoder_1", 0, 0]]}