Skip to content

Pull requests: google/flax

New pull request New
183 Open 2,735 Closed
183 Open 2,735 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Use .format in place of .layout.
#4785 opened Jun 19, 2025 by copybara-service bot Loading…
feat: Add overload signatures to module.apply
#4783 opened Jun 18, 2025 by Daraan Loading…
2 tasks done
1
Fix typo in Flax nnx_basics doc.
#4781 opened Jun 15, 2025 by copybara-service bot Loading…
1
Added a note to gemma guide about model's license consent on kaggle pull ready
#4776 opened Jun 10, 2025 by vfdev-5 Loading…
2
Fixes colab link for nnx docs pull ready
#4775 opened Jun 10, 2025 by vfdev-5 Loading…
Add BatchNorm layers to CNN in MNIST tutorial for improved training stability
#4773 opened Jun 7, 2025 by sanepunk Loading…
2
Fix typo on 'Scale up on multiple devices' guide
#4765 opened May 30, 2025 by DavidUlloa6310 Loading…
2
Add training script for gemma example pull ready
#4761 opened May 26, 2025 by vfdev-5 Loading…
[nnx] mutable array p3 pull ready
#4755 opened May 18, 2025 by cgarciae Loading…
1
[nnx] don't share Rngs
#4754 opened May 17, 2025 by cgarciae Loading…
[nnx] add optax optimizer
#4752 opened May 16, 2025 by cgarciae Loading…
mark shard_map as implemented in transforms guide pull ready
#4738 opened May 5, 2025 by cgarciae Loading…
1
Remove VariableState
#4729 opened Apr 25, 2025 by cgarciae • Draft
1
Fix small typo in performance consideration docs
#4721 opened Apr 20, 2025 by cdelv Loading…
2
FEAT: add LoRALinearGeneral
#4718 opened Apr 19, 2025 by zhengzl18 Loading…
1
convert DenseGeneral to NNX
#4710 opened Apr 17, 2025 by copybara-service bot Loading…
Add attention_bias parameter to MultiHeadDotProductAttention.
#4701 opened Apr 13, 2025 by copybara-service bot Loading…
Added LSTM Model in JAX from scratch
#4690 opened Apr 5, 2025 by Tushaam Loading…
2
bump webhook action version and improve permissions
#4685 opened Apr 4, 2025 by ZacCranko Loading…
Internal change
#4682 opened Apr 4, 2025 by copybara-service bot Loading…
Bump pyo3 from 0.21.2 to 0.24.1 in /flaxlib_src in the cargo group across 1 directory dependencies Pull requests that update a dependency file rust Pull requests that update rust code
#4675 opened Apr 2, 2025 by dependabot bot Loading…
[nnx] refactor remat pull ready
#4662 opened Mar 28, 2025 by cgarciae Loading…
add experimental support for mutable array
#4661 opened Mar 28, 2025 by cgarciae • Draft
add fast_flatten
#4654 opened Mar 26, 2025 by cgarciae Loading…
1
roll back #4646
#4647 opened Mar 24, 2025 by copybara-service bot Loading…
ProTip! Adding no:label will show everything without a label.