| Log | ||
|---|---|---|
| Branch | feat/nnx-post-train-fixes |
|
| Commit | 7f06c99ac |
|
| Date | 2026-04-22T15:46Z | |
| Run ID | feat_nnx_post_train_fixes_20260422 |
|
| Hardware | V6e-32 TPU (32 devices/slice × 1 slice) | |
| Docker image | gcr.io/cienet-cmcs/wanglance-xpk-runner-pre:feat_nnx_post_train_fixes |
|
| NNX flags | pure_nnx=True pure_nnx_decoder=True enable_nnx=True |
26/26 passed.
| Test | Result | Log |
|---|---|---|
01_base |
PASS | log |
02_synthetic |
PASS | log |
03_dropout |
PASS | log |
04_int8 |
PASS | log |
05_fp8 |
PASS | log |
06_grad_accum |
PASS | log |
07_eval |
PASS | log |
08_checkpoint_async_true_save |
PASS | log |
08_checkpoint_async_true_resume |
PASS | log |
09_pdb_lt_1 |
PASS | log |
10_shardy_true |
PASS | log |
10_shardy_false |
PASS | log |
11_optimizer_offload_false |
PASS | log |
11_optimizer_offload_true |
PASS | log |
13_scan_layers_false |
PASS | log |
13_scan_layers_true |
PASS | log |
14_async_ckpt_false_save |
PASS | log |
14_async_ckpt_false_resume |
PASS | log |
15_ocdbt_true_save |
PASS | log |
15_ocdbt_true_resume |
PASS | log |
15_ocdbt_false_save |
PASS | log |
15_ocdbt_false_resume |
PASS | log |
16_zarr3_true_save |
PASS | log |
16_zarr3_true_resume |
PASS | log |
16_zarr3_false_save |
PASS | log |
16_zarr3_false_resume |
PASS | log |
| Log | ||
|---|---|---|
| Logs | /usr/local/google~/maxtext/venv_runs/runs/feat_nnx_post_train_fixes_20260422_123906/nnx/logs/ |
|
| GCS output (checkpoints + profiler) | gs://lance-maxtext/nnx_ckpt_xpk_feat_nnx_post_train_fixes_20260422_123906/ |