Skip to content

Pull requests: google/flax

New pull request New
211 Open 2,884 Closed
211 Open 2,884 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

Support for python 3.14
#5032 opened Oct 15, 2025 by vfdev-5 • Draft
1
2
Added a note in nnx.jit about arg donation
#5031 opened Oct 15, 2025 by vfdev-5 Loading…
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…
Convert linen pytorch guide to nnx
#4999 opened Oct 5, 2025 by samanklesaria Loading…
1
Fix NNX jit static args with in_shardings issue #4989
#4996 opened Oct 3, 2025 by mohsinm-dev Loading…
1
12
Expand ConvTranspose padding documentation
#4990 opened Sep 29, 2025 by samanklesaria Loading…
1
python 3.13 support
#4987 opened Sep 26, 2025 by cgarciae Loading…
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
7
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
[jax:benchmarks] Add tracing/lowering benchmarks for a few flax examples.
#4911 opened Aug 26, 2025 by copybara-service bot Loading…
Refactor ArrayCheckpointHandler to use BasePyTreeCheckpointHandler.
#4902 opened Aug 22, 2025 by copybara-service bot Loading…
1
ProTip! Filter pull requests by the default branch with base:main.