Commit 994338c

committed

feat: migrate pipeline to nnx

1 parent 634672c commit 994338cCopy full SHA for 994338c

2 files changed

src/maxtext/layers
- nnx_decoders.py
- nnx_pipeline.py

Comments

(0)