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

remove fingerprint
#5131 opened Dec 9, 2025 by copybara-service bot Loading…
Add Optimization Cookbook
#5117 opened Nov 28, 2025 by samanklesaria Draft
improve hijax guide
#5115 opened Nov 25, 2025 by cgarciae Draft
Reenabled examples tests
#5098 opened Nov 18, 2025 by vfdev-5 Draft
Refactor nnx.tabulate
#5096 opened Nov 18, 2025 by cgarciae Loading…
Added is_causal mask argument to flax.nnx.dot_product_attention
#5093 opened Nov 17, 2025 by ibbyml Loading…
3 tasks done
Rename sharding_names to sharding_metadata
#5089 opened Nov 13, 2025 by samanklesaria Loading…
Support for python 3.14 pull ready
#5087 opened Nov 13, 2025 by vfdev-5 Loading…
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
fix List and improve Pytree
#5072 opened Nov 4, 2025 by cgarciae Loading…
Use nnx split during tabulate pull ready
#5069 opened Nov 3, 2025 by samanklesaria Loading…
add support for hijax Variables in nnx transforms
#5065 opened Nov 1, 2025 by cgarciae Loading…
Pooling operation
#5057 opened Oct 28, 2025 by jorisSchaller Loading…
1 task done
Box support
#5029 opened Oct 14, 2025 by cgarciae Draft
New set mode
#5028 opened Oct 14, 2025 by chapman20j Loading…
1 task
Make Linen and NNX logical rule deduction align
#5003 opened Oct 7, 2025 by IvyZX Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.