[NNX] Delete Linen (12.1a): collapse dispatch in the core train loop and maxtext_utils - #4683
Draft
ecnal-cienet wants to merge 1 commit into
Draft
[NNX] Delete Linen (12.1a): collapse dispatch in the core train loop and maxtext_utils#4683ecnal-cienet wants to merge 1 commit into
ecnal-cienet wants to merge 1 commit into
