Skip to content

Pull requests: google/flax

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
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

Pooling operation
#5057 opened Oct 28, 2025 by jorisSchaller Loading…
1 task done
Set Mode with Tests
#5056 opened Oct 28, 2025 by chapman20j Loading…
2 tasks
deprecate Variable.value
#5052 opened Oct 24, 2025 by cgarciae Loading…
Box support
#5029 opened Oct 14, 2025 by cgarciae Draft
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…
Expand ConvTranspose padding documentation
#4990 opened Sep 29, 2025 by samanklesaria Loading…
chore: more specific migration guide description
#4977 opened Sep 24, 2025 by emergenz Loading…
1 task done
Hijax Variable pull ready
#4960 opened Sep 18, 2025 by cgarciae Loading…
Add a customizable conv_transpose. pull ready
#4938 opened Sep 11, 2025 by copybara-service bot Loading…
Improve error message with class name.
#4932 opened Sep 9, 2025 by copybara-service bot Loading…
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…
Updates in gemma example
#4913 opened Aug 27, 2025 by vfdev-5 Draft
ProTip! Updated in the last three days: updated:>2025-10-26.