Skip to content

Pull requests: google/flax

New pull request New
Clear current search query, filters, and sorts
196 Open 2,796 Closed
196 Open 2,796 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

add find_duplicates pull ready
#4894 by cgarciae was merged Aug 18, 2025 Loading…
1
Pin sentencepiece version to 0.2.0 to fix head pull ready
#4892 by IvyZX was merged Aug 14, 2025 Loading…
call jax's source_info_util.register_exclusion in flax's traceback_util.register_exclusion
#4887 by copybara-service bot was merged Aug 13, 2025 Loading…
Internal change
#4886 by copybara-service bot was merged Aug 12, 2025 Loading…
Update doc build instructions and clean up unused packages pull ready
#4885 by IvyZX was merged Aug 12, 2025 Loading…
1
Improve docs related with dataclasses pull ready
#4884 by IvyZX was merged Aug 12, 2025 Loading…
1
1
Relax duplicate check to exclude non-string values such as PartitionSpec.UNCONSTRAINED, since those can be repeated.
#4881 by copybara-service bot was merged Aug 14, 2025 Loading…
Update typo in nnx.Optimizer pull ready
#4880 by codinfox was merged Aug 14, 2025 Loading…
1 of 4 tasks
updates to version 0.11.1 pull ready
#4878 by cgarciae was merged Aug 8, 2025 Loading…
Remove the need for Conv and ConvTranspose to know the precise batch size.
#4877 by copybara-service bot was merged Aug 13, 2025 Loading…
add Rngs random helpers pull ready
#4876 by cgarciae was merged Aug 8, 2025 Loading…
2
update and pop for dict pull ready
#4869 by cgarciae was merged Aug 8, 2025 Loading…
simplify nnx_basics pull ready
#4868 by cgarciae was merged Aug 8, 2025 Loading…
9
# Description
#4866 by copybara-service bot was merged Aug 5, 2025 Loading…
4 tasks done
2
Add old property attributes for object->pytree rename.
#4865 by IvyZX was closed Aug 2, 2025 Loading…
Add old property attributes for object->pytree rename.
#4864 by copybara-service bot was merged Aug 2, 2025 Loading…
Pytree and ArrayRef refactor pull ready
#4863 by cgarciae was merged Aug 1, 2025 Loading…
1
Replace jax.sharding.use_mesh with jax.set_mesh. jax.set_mesh can act as a global setter or a context manager.
#4862 by copybara-service bot was merged Aug 1, 2025 Loading…
ArrayRef refactor pull ready
#4861 by cgarciae was closed Aug 8, 2025 Loading…
1
Fix string key preservation in replace_by_pure_dict pull ready
#4860 by mohsinm-dev was merged Aug 12, 2025 Loading…
2 of 4 tasks
1
1
miscellaneous improvements pull ready
#4859 by cgarciae was merged Jul 31, 2025 Loading…
Fix broken contributing documentation link pull ready
#4855 by mohsinm-dev was merged Aug 12, 2025 Loading…
2 of 4 tasks
1
2
0.11.0 migration guide pull ready
#4854 by cgarciae was merged Jul 29, 2025 Loading…
6
Fix ToLinen docstring return description pull ready
#4852 by mohsinm-dev was merged Aug 11, 2025 Loading…
1 of 4 tasks
1
2
[nnx] add mutable array support in update pull ready
#4851 by cgarciae was merged Jul 26, 2025 Loading…
ProTip! Add no:assignee to see everything that’s not assigned.