Skip to content

Pull requests: google/flax

New pull request New
218 Open 2,927 Closed
218 Open 2,927 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

Reenabled examples tests
#5098 opened Nov 18, 2025 by vfdev-5 • Draft
1
Fixed deprecated .value usage failing CI tests
#5097 opened Nov 18, 2025 by vfdev-5 Loading…
fix nnx.tabulate
#5096 opened Nov 18, 2025 by cgarciae Loading…
update jax minver to 0.8.1
#5095 opened Nov 17, 2025 by cgarciae Loading…
Allow substring matching in nnx.PathContains
#5094 opened Nov 17, 2025 by thijs-vanweezel Loading…
4 tasks done
1
5
Added is_causal mask argument to flax.nnx.dot_product_attention
#5093 opened Nov 17, 2025 by ibbyml Loading…
3 tasks done
5
Fixed dataset path in train tests in examples
#5092 opened Nov 17, 2025 by vfdev-5 • Draft
Rename sharding_names to sharding_metadata
#5089 opened Nov 13, 2025 by samanklesaria Loading…
Updated flax_gspmd/bridge_guide to use kernel_metadata, make_mesh and sharding_names
#5088 opened Nov 13, 2025 by vfdev-5 Loading…
1
Support for python 3.14
#5087 opened Nov 13, 2025 by vfdev-5 • Draft
1
10
Updated pyproject.toml
#5086 opened Nov 13, 2025 by vfdev-5 Loading…
Migrate VAE example to Flax NNX
#5077 opened Nov 7, 2025 by sanepunk Loading…
5 tasks done
23
fix List and improve Pytree
#5072 opened Nov 4, 2025 by cgarciae Loading…
2
Fallback to non-jit version of methods in tabulate if necessary pull ready
#5069 opened Nov 3, 2025 by samanklesaria Loading…
15
add support for hijax Variables in nnx transforms
#5065 opened Nov 1, 2025 by cgarciae Loading…
Update mnist example to use NNX (clone of #5002) pull ready
#5064 opened Oct 31, 2025 by samanklesaria Loading…
1
1
Pooling operation
#5057 opened Oct 28, 2025 by jorisSchaller Loading…
1 task done
1
13
Add split and key methods to RngStream and Rngs pull ready
#5047 opened Oct 23, 2025 by samanklesaria Loading…
1
5
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…
chore: more specific migration guide description
#4977 opened Sep 24, 2025 by emergenz Loading…
1 task done
ProTip! Exclude everything labeled bug with -label:bug.