Commit ee386ed

committed

Migrate Pipeline to use NNX module.

1 parent bac545a commit ee386edCopy full SHA for ee386ed

6 files changed

+2818

-590

lines changed

src/MaxText/layers
tests
- pipeline_parallelism_test.py
- pipeline_parallelism_test_linen.py

6 files changed

+2818

-590

lines changed

Comments

(0)