Skip to content

Pull requests: google/flax

New pull request New
212 Open 2,901 Closed
212 Open 2,901 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

Pooling operation
#5057 opened Oct 28, 2025 by jorisSchaller Loading…
1 task done
1
Set Mode with Tests
#5056 opened Oct 28, 2025 by chapman20j Loading…
2 tasks
Fix bound method auto-unbinding for NNX transforms
#5055 opened Oct 28, 2025 by mohsinm-dev Loading…
5
deprecate Variable.value
#5052 opened Oct 24, 2025 by cgarciae Loading…
1
Add split and key methods to RngStream and Rngs
#5047 opened Oct 23, 2025 by samanklesaria Loading…
1
1
Add REINFORCE policy gradient example for CartPole using Flax NNX
#5036 opened Oct 19, 2025 by sanepunk Loading…
6
Box support
#5029 opened Oct 14, 2025 by cgarciae • Draft
1
New set mode
#5028 opened Oct 14, 2025 by chapman20j Loading…
1 task
[flax:examples:vae] Small linter fixes.
#5014 opened Oct 9, 2025 by copybara-service bot Loading…
[flax:examples:gemma] Fixing linter errors.
#5013 opened Oct 9, 2025 by copybara-service bot Loading…
[flax:examples:wmt] Small linter fixes.
#5012 opened Oct 9, 2025 by copybara-service bot Loading…
Make Linen and NNX logical rule deduction align
#5003 opened Oct 7, 2025 by IvyZX Loading…
[flax:examples:mnist] Update mnist example to use NNX.
#5002 opened Oct 6, 2025 by copybara-service bot Loading…
Fix NNX jit static args with in_shardings issue #4989
#4996 opened Oct 3, 2025 by mohsinm-dev Loading…
1
31
Expand ConvTranspose padding documentation
#4990 opened Sep 29, 2025 by samanklesaria Loading…
1
chore: more specific migration guide description
#4977 opened Sep 24, 2025 by emergenz Loading…
1 task done
Update tutorial examples to thread explicit RNGs
#4975 opened Sep 23, 2025 by samanklesaria Loading…
1
8
Hijax Variable pull ready
#4960 opened Sep 18, 2025 by cgarciae Loading…
1
Add a customizable conv_transpose. pull ready
#4938 opened Sep 11, 2025 by copybara-service bot Loading…
1
Don't pass linen_meta_type argument when creating AxisMetadata subclasses.
#4933 opened Sep 9, 2025 by copybara-service bot Loading…
1
Improve error message with class name.
#4932 opened Sep 9, 2025 by copybara-service bot Loading…
1
Set mode
#4927 opened Sep 5, 2025 by cgarciae Loading…
adds recursive_map
#4926 opened Sep 5, 2025 by copybara-service bot Loading…
Add intermediate value captures based off hijax.Box
#4925 opened Sep 4, 2025 by IvyZX Loading…
6
Updates in gemma example
#4913 opened Aug 27, 2025 by vfdev-5 • Draft
1
ProTip! Updated in the last three days: updated:>2025-10-26.