MaxView

Timing Test Reference Updates

Post-Training Test Results (NNX)

Environment

Log
Branch feat/nnx-set-defaults-true
Commit 73213e044
Date 2026-04-25T15:35Z
Run ID feat_nnx_set_defaults_true_20260425_150421
NNX flags pure_nnx=True enable_nnx=True pure_nnx_decoder=True
Hardware V6e-32 TPU (32 devices/slice × 1 slice)
Docker image gcr.io/cienet-cmcs/wanglance-xpk-runner-pt:feat_nnx_set_defaults_true

Summary

0/7 passed.

Test Result Log
01_sft_smoke FAIL log
02_sft_nnx_ckpt FAIL log
03_sft_linen_ckpt FAIL log
07_distill_smoke FAIL log
04_rl_grpo_smoke FAIL log
05_rl_gspo_smoke FAIL log
06_rl_grpo_functional FAIL log

Output Paths

Log
Logs /usr/local/google~/maxtext/venv_runs/runs/feat_nnx_set_defaults_true_20260425_150421/nnx/logs/
GCS checkpoints gs://lance-maxtext/pt_ckpt_xpk_feat_nnx_set_defaults_true_20260425_150421/
NNX seed ckpt (PT-02 NNX) gs://lance-maxtext/nnx_ckpt_feat_nnx_trainstate_and_training_loop_20260411_044231/nnx_ckpt_feat_nnx_trainstate_and_training_loop_20260411_044231/nnx_feat_nnx_trainstate_and_training_loop_20260411_044231_08_checkpoint_async_true/checkpoints/9/items
Linen seed ckpt (PT-02/03) gs://lance-maxtext/pt_seed_ckpts/pt_seed_ckpts/pt_seed_ckpt_gpt352k_linen/checkpoints/9/items
Teacher ckpt (PT-07) gs://lance-maxtext/pt_seed_ckpts/pt_seed_ckpts/pt_seed_ckpt_gpt352k_v32k_linen/checkpoints/4/items
RL ckpt (PT-06) gs://lance-maxtext/rl_ckpt_llama31_8b/rl_ckpt_llama31_8b/0/items