Fix sharding rank mismatch for LoRA weights and optimizer states in grpo_gemma.ipynb notebook
#4231
| Job | Run time |
|---|---|
| 11s | |
| 1m 21s | |
| 13m 6s | |
| 0s | |
| 4s | |
| 14m 42s |
grpo_gemma.ipynb notebook
#4231
| Job | Run time |
|---|---|
| 11s | |
| 1m 21s | |
| 13m 6s | |
| 0s | |
| 4s | |
| 14m 42s |