SPMD for initializing model using nnx.jit
Open
Priority: P1 - soon
- Dominant language
- Jupyter Notebook
- Stars
- 7.3k
- Forks
- 833
- Avg merge
- 5h 11m
- Merged PRs (30d)
- 5
Description
I would like to ask an example for initializing and updating a model using nnx.jit with SPMD.
Is there any relevant example?
Contributor guide
Assessment
This issue has not been assessed yet.