Fix sharding rank mismatch for LoRA weights and optimizer states in grpo_gemma.ipynb notebook
#4231
rajasekharporeddybuild_and_test_tunix.yml
on: pull_request
Build tunix package
/
Build Tunix wheel
11s
tunix_tpu_unit_tests
/
run_prod
13m 6s
tunix_tpu_unit_tests
/
run_dev
0s
Notify failed build
4s
Artifacts
Produced during runtime
| Name | Size | Digest | |
|---|---|---|---|
|
tunix-wheel
|
356 KB |
sha256:8261f3b8833ed3bd69e8a150ed0d50a69a3830dc24aae1abad6f027d21fec190
|
|