-
Notifications
You must be signed in to change notification settings - Fork 677
Pull requests: google/flax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add Sow Config to from_params constructor.
#4599
opened Mar 4, 2025 by
copybara-service
bot
Loading…
Fix position and name of Post Attention Norm.
#4598
opened Mar 4, 2025 by
copybara-service
bot
Loading…
Add option to load checkpoints with transposed Gating Einsum.
#4597
opened Mar 4, 2025 by
copybara-service
bot
Loading…
Add configurable Query Pre Attention scalar.
#4595
opened Mar 4, 2025 by
copybara-service
bot
Loading…
Adding nnx Gemma2-2b (including overall fixes) to examples/gemma
#4587
opened Feb 28, 2025 by
mdda
Loading…
Canonicalize PartitionSpec so that we can delete ParsedPartitionSpec
#4577
opened Feb 26, 2025 by
copybara-service
bot
Loading…
Dramatically speed up sampling compilation time
#4574
opened Feb 25, 2025 by
copybara-service
bot
Loading…
Dramatically speed up sampling compilation time
#4573
opened Feb 25, 2025 by
copybara-service
bot
Loading…
Add some methods on Sequential container
#4565
opened Feb 22, 2025 by
Ruhaan838
Loading…
2 of 3 tasks
Fixes #1925 non-str dict keys not suppoted in module state
#4563
opened Feb 21, 2025 by
muhrin
Loading…
3 of 4 tasks
Add REFLECT padding to convolution layer
#4553
opened Feb 19, 2025 by
sarlinpe
Loading…
1 of 4 tasks
Improve error message for when nnx.Modules use jax or numpy arrays as leaf values
#4492
opened Jan 20, 2025 by
RaghuSpaceRajan
Loading…
[linen] Linesearch (and lbfgs) support for TrainState
#4471
opened Jan 6, 2025 by
emiresenov
Loading…
3 of 4 tasks
Update NNX
State.state.merge method docs in statelib.py
#4453
opened Dec 17, 2024 by
8bitmp3
Loading…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.