| Log | ||
|---|---|---|
| Branch | feat/nnx-trainstate-and-training-loop |
|
| Commit | e27fc1e97 |
|
| Date | 2026-04-21T11:40Z | |
| Run ID | feat_nnx_trainstate_and_training_loop_20260421 |
|
| Hardware | V6e-32 TPU (32 devices/slice × 1 slice) | |
| Docker image | gcr.io/cienet-cmcs/wanglance-xpk-runner-pre:feat_nnx_trainstate_and_training_loop |
|
| 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/feat_nnx_trainstate_and_training_loop_20260421_083238/nnx/logs/ |
|
| GCS output (checkpoints + profiler) | gs://lance-maxtext/nnx_ckpt_xpk_feat_nnx_trainstate_and_training_loop_20260421_083238/ |