NNX migration prep (3/N): TrainState, model creation, and end-to-end training loop#3500
Draft
ecnal-cienet wants to merge 3 commits intomainfrom
Draft
NNX migration prep (3/N): TrainState, model creation, and end-to-end training loop#3500ecnal-cienet wants to merge 3 commits intomainfrom
ecnal-cienet wants to merge 3 commits intomainfrom
Commits
Commits on Apr 2, 2026
- authored andcommitted
- authored andcommitted
- committed
