diff --git a/middleout-lattice.ipynb b/middleout-lattice.ipynb
new file mode 100644
index 0000000000..3e89513d68
--- /dev/null
+++ b/middleout-lattice.ipynb
@@ -0,0 +1 @@
+{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.12.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"nvidiaTeslaT4","dataSources":[],"dockerImageVersionId":31329,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"!pip install range-coder && python middleout-lattice/scripts/train_lightning.py --model-config small --epochs 10 --batch-size 32","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2026-05-08T10:21:05.676162Z","iopub.execute_input":"2026-05-08T10:21:05.677050Z"}},"outputs":[{"name":"stdout","text":"Requirement already satisfied: range-coder in /usr/local/lib/python3.12/dist-packages (1.1)\n============================================================\nLIGHTNING AI NEURAL COMPRESSION TRAINING\n============================================================\nData dir: /kaggle/working/middleout-lattice/data\nOutput: /kaggle/working/middleout-lattice/checkpoints/model.pt\nModel config: small\nEpochs: 10\nBatch size: 32\nGutenberg count: 5000\nGPU available: True\nGPU: Tesla T4\nVRAM: 15.6 GB\n============================================================\n\n[Step 1/3] Preparing training data...\n\n============================================================\nDOWNLOADING TRAINING DATA\n============================================================\n\nDownloading 5000 Gutenberg books (50 parallel workers)...\n100%|██████████████████████████████████████| 5000/5000 [00:44<00:00, 112.12it/s]\n✓ Gutenberg: 5000 books\n\nDownloading 35 code repositories (20 parallel workers)...\n100%|███████████████████████████████████████████| 35/35 [00:46<00:00,  1.33s/it]\n✓ Code repos: 33 archives\n\nDownloading 10 JSON/CSV/MD datasets (15 parallel workers)...\n100%|███████████████████████████████████████████| 10/10 [00:00<00:00, 34.50it/s]\n✓ JSON/CSV datasets: 8 files\n\nDownloading 5 web files (10 parallel workers)...\n100%|█████████████████████████████████████████████| 5/5 [00:00<00:00, 90.10it/s]\n✓ Web files (HTML/CSS/JS): 5 files\n\nDownloading 4 config files (8 parallel workers)...\n100%|█████████████████████████████████████████████| 4/4 [00:00<00:00, 37.41it/s]\n✓ Config files: 3 files\n\nDownloading 9 markdown docs (12 parallel workers)...\n100%|████████████████████████████████████████████| 9/9 [00:00<00:00, 104.56it/s]\n✓ Markdown docs: 6 files\n\nGenerating sample log files...\n✓ Log files: 4 files\n\nGenerating binary test files...\n✓ Binary files: 40 files\n\nGenerating 2000 synthetic files...\n100%|█████████████████████████████████████| 2000/2000 [00:00<00:00, 9318.70it/s]\n✓ Synthetic files: 2000 files\n\nDownloading tinyshakespeare...\n✓ TinyShakespeare: 1 file\n\nExtracting archives...\n100%|███████████████████████████████████████████| 33/33 [02:30<00:00,  4.57s/it]\n  0%|                                                     | 0/2 [00:00<?, ?it/s]/kaggle/working/middleout-lattice/scripts/prepare_data.py:586: DeprecationWarning: Python 3.14 will, by default, filter extracted tar archives and reject files or modify their metadata. Use the filter argument to control this behavior.\n  tf.extractall(extract_dir)\n100%|████████████████████████████████████████████| 2/2 [00:00<00:00, 442.16it/s]\n\n============================================================\nTRAINING DATA SUMMARY\n============================================================\nText files (.txt):     26,800\nJSON files:            8,653\nPython files (.py):    25,925\nJavaScript (.js):      62,448\nTypeScript (.ts):      42,495\nCSS files:             2,504\nHTML files:            2,766\nMarkdown (.md):        11,111\nCSV files:             411\nXML files:             638\nYAML files:            8,623\nTOML files:            1,160\nINI files:             215\nSQL files:             1,040\nLog files:             39\nBinary files:          206\nOther files:           639,092\n------------------------------------------------------------\nTotal size:            10,598.6 MB\nTotal size:            10.60 GB\n============================================================\n\nSummary saved to: /kaggle/working/middleout-lattice/data/data_summary.json\n\n[Step 2/3] Training model...\nTraining config: TrainingConfig(data_dir=PosixPath('/kaggle/working/middleout-lattice/data'), output=PosixPath('/kaggle/working/middleout-lattice/checkpoints/model.pt'), model_config='small', epochs=10, batch_size=32, lr=0.0003, warmup_steps=500, max_steps=-1, grad_accum=1, seq_len=-1, checkpoint_every=500, resume=None, device='cuda', use_multi_gpu=True)\nUsing device: cuda\nGPUs available: 2x Tesla T4 (15.6 GB each)\nAuto-adjusted seq_len: -1 → 1024 (model context_len)\nModel config: ModelConfig(vocab_size=256, context_len=1024, embed_dim=256, n_layers=6, n_heads=4, ff_dim=1024, dropout=0.1, bias=True)\nEstimated params: 5,118,208\nUsing 2 GPUs with DataParallel\nActual params: 6,707,968\nEpoch 1: 499it [04:50,  1.70it/s, loss=3.62, lr=0.0003, ppl=37.3]  \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_500.pt\nEpoch 1: 999it [09:46,  1.68it/s, loss=2.95, lr=0.0003, ppl=19.1]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_1000.pt\nEpoch 1: 1499it [14:42,  1.68it/s, loss=2.88, lr=0.0003, ppl=17.9]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_1500.pt\nEpoch 1: 1999it [19:41,  1.70it/s, loss=2.85, lr=0.0003, ppl=17.3]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_2000.pt\nEpoch 1: 2499it [24:37,  1.70it/s, loss=2.73, lr=0.0003, ppl=15.3]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_2500.pt\nEpoch 1: 2999it [29:35,  1.70it/s, loss=2.72, lr=0.0003, ppl=15.2]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_3000.pt\nEpoch 1: 3499it [34:31,  1.69it/s, loss=2.7, lr=0.0003, ppl=14.9] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_3500.pt\nEpoch 1: 3999it [39:28,  1.70it/s, loss=2.69, lr=0.0003, ppl=14.7]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_4000.pt\nEpoch 1: 4499it [44:27,  1.66it/s, loss=2.66, lr=0.0003, ppl=14.3]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_4500.pt\nEpoch 1: 4999it [49:22,  1.69it/s, loss=2.6, lr=0.0003, ppl=13.5] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_5000.pt\nEpoch 1: 5499it [54:19,  1.68it/s, loss=2.57, lr=0.0003, ppl=13]  \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_5500.pt\nEpoch 1: 5999it [59:16,  1.69it/s, loss=2.53, lr=0.0003, ppl=12.6]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_6000.pt\nEpoch 1: 6412it [1:03:21,  1.68it/s, loss=2.51, lr=0.0003, ppl=12.3]Warning: failed to read /kaggle/working/middleout-lattice/data/code/nextjs/next.js-canary/test/e2e/app-dir/app-routes/app/api/hello.json: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/nextjs/next.js-canary/test/e2e/app-dir/app-routes/app/api/hello.json'\nEpoch 1: 6499it [1:04:14,  1.68it/s, loss=2.5, lr=0.0003, ppl=12.2] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_6500.pt\nEpoch 1: 6999it [1:09:12,  1.68it/s, loss=2.45, lr=0.0003, ppl=11.5]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_7000.pt\nEpoch 1: 7499it [1:14:10,  1.68it/s, loss=2.38, lr=0.0003, ppl=10.8]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_7500.pt\nEpoch 1: 7999it [1:19:09,  1.67it/s, loss=2.33, lr=0.0003, ppl=10.2]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_8000.pt\nEpoch 1: 8499it [1:24:07,  1.68it/s, loss=2.25, lr=0.0003, ppl=9.47]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_8500.pt\nEpoch 1: 8999it [1:29:05,  1.67it/s, loss=2.18, lr=0.0003, ppl=8.85]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_9000.pt\nEpoch 1: 9499it [1:34:02,  1.68it/s, loss=2.15, lr=0.0003, ppl=8.55]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_9500.pt\nEpoch 1: 9999it [1:39:00,  1.67it/s, loss=2.1, lr=0.0003, ppl=8.19] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_10000.pt\nEpoch 1: 10499it [1:43:57,  1.70it/s, loss=2.07, lr=0.0003, ppl=7.92]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_10500.pt\nEpoch 1: 10999it [1:48:53,  1.70it/s, loss=2.04, lr=0.0003, ppl=7.68]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_11000.pt\nEpoch 1: 11499it [1:53:50,  1.68it/s, loss=2.01, lr=0.0003, ppl=7.45]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_11500.pt\nEpoch 1: 11999it [1:58:46,  1.70it/s, loss=1.98, lr=0.0003, ppl=7.23]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_12000.pt\nEpoch 1: 12499it [2:03:43,  1.70it/s, loss=1.95, lr=0.0003, ppl=7.05]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_12500.pt\nEpoch 1: 12999it [2:08:40,  1.69it/s, loss=1.92, lr=0.0003, ppl=6.84]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_13000.pt\nEpoch 1: 13499it [2:13:37,  1.70it/s, loss=1.89, lr=0.0003, ppl=6.6] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_13500.pt\nEpoch 1: 13999it [2:18:33,  1.68it/s, loss=1.86, lr=0.0003, ppl=6.45]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_14000.pt\nEpoch 1: 14499it [2:23:31,  1.68it/s, loss=1.84, lr=0.0003, ppl=6.29]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_14500.pt\nEpoch 1: 14999it [2:28:27,  1.70it/s, loss=1.81, lr=0.0003, ppl=6.14]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_15000.pt\nEpoch 1: 15499it [2:33:24,  1.70it/s, loss=1.79, lr=0.0003, ppl=6]   \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_15500.pt\nEpoch 1: 15999it [2:38:20,  1.69it/s, loss=1.77, lr=0.0003, ppl=5.85]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_16000.pt\nEpoch 1: 16499it [2:43:17,  1.70it/s, loss=1.74, lr=0.0003, ppl=5.71]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_16500.pt\nEpoch 1: 16999it [2:48:15,  1.68it/s, loss=1.72, lr=0.0003, ppl=5.59]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_17000.pt\nEpoch 1: 17499it [2:53:13,  1.69it/s, loss=1.7, lr=0.0003, ppl=5.5]  \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_17500.pt\nEpoch 1: 17999it [2:58:09,  1.69it/s, loss=1.68, lr=0.0003, ppl=5.38]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_18000.pt\nEpoch 1: 18499it [3:03:05,  1.69it/s, loss=1.67, lr=0.0003, ppl=5.3] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_18500.pt\nEpoch 1: 18999it [3:08:02,  1.68it/s, loss=1.66, lr=0.0003, ppl=5.25]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_19000.pt\nEpoch 1: 19499it [3:12:58,  1.70it/s, loss=1.64, lr=0.0003, ppl=5.17]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_19500.pt\nEpoch 1: 19999it [3:17:53,  1.70it/s, loss=1.63, lr=0.0003, ppl=5.1] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_20000.pt\nEpoch 1: 20067it [3:18:32,  1.70it/s, loss=1.63, lr=0.0003, ppl=5.09]Warning: failed to read /kaggle/working/middleout-lattice/data/code/nextjs/next.js-canary/test/e2e/app-dir/app-routes/app/static/[slug]/data.json: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/nextjs/next.js-canary/test/e2e/app-dir/app-routes/app/static/[slug]/data.json'\nEpoch 1: 20499it [3:22:48,  1.70it/s, loss=1.62, lr=0.0003, ppl=5.04]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_20500.pt\nEpoch 1: 20999it [3:27:47,  1.68it/s, loss=1.61, lr=0.0003, ppl=4.99]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_21000.pt\nEpoch 1: 21499it [3:32:44,  1.68it/s, loss=1.6, lr=0.0003, ppl=4.94] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_21500.pt\nEpoch 1: 21999it [3:37:41,  1.70it/s, loss=1.59, lr=0.0003, ppl=4.89]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_22000.pt\nEpoch 1: 22499it [3:42:38,  1.70it/s, loss=1.58, lr=0.0003, ppl=4.83]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_22500.pt\nEpoch 1: 22999it [3:47:33,  1.68it/s, loss=1.56, lr=0.0003, ppl=4.77]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_23000.pt\nEpoch 1: 23499it [3:52:31,  1.69it/s, loss=1.55, lr=0.0003, ppl=4.73]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_23500.pt\nEpoch 1: 23999it [3:57:28,  1.70it/s, loss=1.54, lr=0.0003, ppl=4.68]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_24000.pt\nEpoch 1: 24499it [4:02:25,  1.66it/s, loss=1.53, lr=0.0003, ppl=4.62]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_24500.pt\nEpoch 1: 24999it [4:07:24,  1.70it/s, loss=1.52, lr=0.0003, ppl=4.59]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_25000.pt\nEpoch 1: 25499it [4:12:20,  1.68it/s, loss=1.51, lr=0.0003, ppl=4.54]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_25500.pt\nEpoch 1: 25608it [4:13:25,  1.68it/s, loss=1.51, lr=0.0003, ppl=4.53]Warning: failed to read /kaggle/working/middleout-lattice/data/code/llvm/llvm-project-main/libcxx/test/std/time/time.cal/time.cal.md: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/llvm/llvm-project-main/libcxx/test/std/time/time.cal/time.cal.md'\nEpoch 1: 25999it [4:17:18,  1.67it/s, loss=1.5, lr=0.0003, ppl=4.5]  \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_26000.pt\nEpoch 1: 26499it [4:22:17,  1.69it/s, loss=1.49, lr=0.0003, ppl=4.45]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_26500.pt\nEpoch 1: 26999it [4:27:14,  1.67it/s, loss=1.48, lr=0.0003, ppl=4.41]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_27000.pt\nEpoch 1: 27235it [4:29:35,  1.67it/s, loss=1.48, lr=0.0003, ppl=4.4]Warning: failed to read /kaggle/working/middleout-lattice/data/code/webpack/webpack-main/test/configCases/css/prefer-relative/node_modules/foo.css: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/webpack/webpack-main/test/configCases/css/prefer-relative/node_modules/foo.css'\nEpoch 1: 27499it [4:32:14,  1.67it/s, loss=1.48, lr=0.0003, ppl=4.38]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_27500.pt\nEpoch 1: 27999it [4:37:48,  1.63it/s, loss=1.47, lr=0.0003, ppl=4.34]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_28000.pt\nEpoch 1: 28127it [4:39:08,  1.66it/s, loss=1.47, lr=0.0003, ppl=4.33]Warning: failed to read /kaggle/working/middleout-lattice/data/code/webpack/webpack-main/test/configCases/css/prefer-relative-css-import/node_modules/bar.modules.css: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/webpack/webpack-main/test/configCases/css/prefer-relative-css-import/node_modules/bar.modules.css'\nEpoch 1: 28499it [4:42:51,  1.68it/s, loss=1.46, lr=0.0003, ppl=4.29]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_28500.pt\nEpoch 1: 28999it [4:47:49,  1.67it/s, loss=1.44, lr=0.0003, ppl=4.23]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_29000.pt\nEpoch 1: 29262it [4:50:25,  1.70it/s, loss=1.44, lr=0.0003, ppl=4.21]Warning: failed to read /kaggle/working/middleout-lattice/data/code/llvm/llvm-project-main/libcxx/test/selftest/pass.cpp: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/llvm/llvm-project-main/libcxx/test/selftest/pass.cpp'\nEpoch 1: 29499it [4:52:45,  1.69it/s, loss=1.43, lr=0.0003, ppl=4.2] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_29500.pt\nEpoch 1: 29999it [4:57:41,  1.70it/s, loss=1.43, lr=0.0003, ppl=4.17]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_30000.pt\nEpoch 1: 30499it [5:02:38,  1.70it/s, loss=1.42, lr=0.0003, ppl=4.16]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_30500.pt\nEpoch 1: 30999it [5:07:35,  1.69it/s, loss=1.42, lr=0.0003, ppl=4.13]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_31000.pt\nEpoch 1: 31499it [5:12:32,  1.67it/s, loss=1.41, lr=0.0003, ppl=4.1] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_31500.pt\nEpoch 1: 31999it [5:17:28,  1.70it/s, loss=1.4, lr=0.0003, ppl=4.07] \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_32000.pt\nEpoch 1: 32499it [5:22:25,  1.67it/s, loss=1.4, lr=0.0003, ppl=4.04]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_32500.pt\nEpoch 1: 32999it [5:27:23,  1.68it/s, loss=1.39, lr=0.0003, ppl=4.03]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_33000.pt\nEpoch 1: 33499it [5:32:20,  1.68it/s, loss=1.39, lr=0.0003, ppl=4]   \nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_33500.pt\nEpoch 1: 33999it [5:37:15,  1.68it/s, loss=1.38, lr=0.0003, ppl=3.99]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_34000.pt\nEpoch 1: 34300it [5:40:45,  1.12s/it, loss=1.38, lr=0.0003, ppl=3.97]Warning: failed to read /kaggle/working/middleout-lattice/data/code/llvm/llvm-project-main/libcxx/test/selftest/verify.cpp: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/llvm/llvm-project-main/libcxx/test/selftest/verify.cpp'\nEpoch 1: 34499it [5:43:40,  1.36it/s, loss=1.38, lr=0.0003, ppl=3.96]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_34500.pt\nEpoch 1: 34999it [5:49:32,  1.67it/s, loss=1.37, lr=0.0003, ppl=3.94]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_35000.pt\nEpoch 1: 35482it [5:54:18,  1.68it/s, loss=1.36, lr=0.0003, ppl=3.91]Warning: failed to read /kaggle/working/middleout-lattice/data/code/webpack/webpack-main/examples/build-http/webpack.lock.data/https_cdn.esm.sh: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/webpack/webpack-main/examples/build-http/webpack.lock.data/https_cdn.esm.sh'\nEpoch 1: 35499it [5:54:29,  1.67it/s, loss=1.36, lr=0.0003, ppl=3.91]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_35500.pt\nEpoch 1: 35999it [5:59:28,  1.68it/s, loss=1.36, lr=0.0003, ppl=3.89]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_36000.pt\nEpoch 1: 36499it [6:04:26,  1.70it/s, loss=1.35, lr=0.0003, ppl=3.87]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_36500.pt\nEpoch 1: 36660it [6:06:01,  1.68it/s, loss=1.35, lr=0.0003, ppl=3.87]Warning: failed to read /kaggle/working/middleout-lattice/data/code/webpack/webpack-main/test/configCases/css/prefer-relative-css-import/node_modules/foo.css: [Errno 21] Is a directory: '/kaggle/working/middleout-lattice/data/code/webpack/webpack-main/test/configCases/css/prefer-relative-css-import/node_modules/foo.css'\nEpoch 1: 36999it [6:09:24,  1.68it/s, loss=1.35, lr=0.0003, ppl=3.86]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_37000.pt\nEpoch 1: 37499it [6:14:21,  1.69it/s, loss=1.35, lr=0.0003, ppl=3.84]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_37500.pt\nEpoch 1: 37999it [6:19:17,  1.69it/s, loss=1.34, lr=0.0003, ppl=3.83]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_38000.pt\nEpoch 1: 38499it [6:24:24,  1.57it/s, loss=1.34, lr=0.0003, ppl=3.81]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_38500.pt\nEpoch 1: 38999it [6:29:24,  1.68it/s, loss=1.33, lr=0.0003, ppl=3.78]\nSaved checkpoint to /kaggle/working/middleout-lattice/checkpoints/checkpoint_step_39000.pt\nEpoch 1: 39113it [6:30:32,  1.67it/s, loss=1.33, lr=0.0003, ppl=3.78]","output_type":"stream"}],"execution_count":null},{"cell_type":"code","source":"python scripts/export_onnx.py \\\n  --model /kaggle/working/checkpoints/checkpoint_step_32500.pt \\\n  --output /kaggle/working/checkpoints/model.onnx \\\n  --optimize","metadata":{"trusted":true},"outputs":[],"execution_count":null}]}
\ No newline at end of file
diff --git a/src/google/adk/tools/agent_tool.py b/src/google/adk/tools/agent_tool.py
index 8a43eb6311..9fe57d6632 100644
--- a/src/google/adk/tools/agent_tool.py
+++ b/src/google/adk/tools/agent_tool.py
@@ -272,6 +272,16 @@ async def run_async(
     merged_text = '\n'.join(
         p.text for p in last_content.parts if p.text and not p.thought
     )
+
+    # Fall back to code_execution_result output if merged_text is empty
+    # This handles inner code-executor agents that return only
+    # executable_code and code_execution_result (no text part)
+    if not merged_text or not merged_text.strip():
+      for p in last_content.parts:
+        if p.code_execution_result and p.code_execution_result.output:
+          merged_text = p.code_execution_result.output
+          break
+
     output_schema = _get_output_schema(self.agent)
     if output_schema:
       tool_result = validate_schema(output_schema, merged_text)