[NNX] NNX migration prep (7/N): NNX-native MaxEngine inference#3821

Draft

ecnal-cienet wants to merge 6 commits into

mainfrom

feat/nnx-native-maxengine

Commits on May 8, 2026

NNX: add TrainState, model creation utilities, and training loop support
ecnal-cienet
committed
NNX migration prep (4/N): sharding tools + post-training bugfixes
ecnal-cienet
committed
NNX migration prep (4.5/N): Linen<->NNX checkpoint converter
ecnal-cienet
committed
NNX: correctness fixes, enable feature paths, and vocab tiling on NNX
ecnal-cienet
committed
NNX: native DPO (TrainStateNNX.reference_model + dpo_loss_fn_nnx)
ecnal-cienet
committed
NNX: native MaxEngine inference (drop route-to-Linen path in maxengine.py)
ecnal-cienet
committed