Skip to content

update jax/flax version#274

Merged
susanbao merged 5 commits intomainfrom
sanbao/flax
Oct 22, 2025
Merged

update jax/flax version#274
susanbao merged 5 commits intomainfrom
sanbao/flax

Conversation

@susanbao
Copy link
Copy Markdown
Collaborator

@susanbao susanbao commented Oct 21, 2025

Flax<0.12.0 caused the "LookupError: Unrecognized config option: flax_always_shard_variable". Update the requirement for Flax and corresponding JAX to avoid this problem.

@susanbao susanbao requested a review from entrpn October 21, 2025 22:57
@github-actions
Copy link
Copy Markdown

@susanbao susanbao merged commit ae2a542 into main Oct 22, 2025
1 of 3 checks passed
@susanbao susanbao deleted the sanbao/flax branch December 11, 2025 23:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants