GitHub Actions workflow [CI - with JAX nightly #514](https://github.com/google/flax/actions/runs/26513374963) failed. Event: schedule Branch: [main](https://github.com/google/flax/tree/main) Commit: [0f128b5141a7ab40df430d280bc86d3929ccb1ce](https://github.com/google/flax/commit/0f128b5141a7ab40df430d280bc86d3929ccb1ce) <sup><i>Created by [jayqi/failed-build-issue-action](https://github.com/jayqi/failed-build-issue-action)</i></sup>
GitHub Actions workflow CI - with JAX nightly #514 failed.
Event: schedule
Branch: main
Commit: 0f128b5141a7ab40df430d280bc86d3929ccb1ce
Created by jayqi/failed-build-issue-action