Skip to content

Pull requests: google/flax

New pull request New
171 Open 2,679 Closed
171 Open 2,679 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Test runner ubuntu upgrade 24.04
#4659 opened Mar 27, 2025 by IvyZX Loading…
Update runners to ubuntu-24.04 from deprecated ubuntu-20.04 label
#4658 opened Mar 26, 2025 by google-admin Loading…
Fix lazy_init typo
#4657 opened Mar 26, 2025 by IvyZX Loading…
1
v0.10.5
#4656 opened Mar 26, 2025 by cgarciae Loading…
add fast_flatten
#4654 opened Mar 26, 2025 by cgarciae Loading…
Added identity export from JAX
#4652 opened Mar 25, 2025 by jlperla Loading…
1
1
roll back #4646
#4647 opened Mar 24, 2025 by copybara-service bot Loading…
Apply vmap jnp.pad over the batch in REFLECT convolution mode
#4640 opened Mar 20, 2025 by sarlinpe Loading…
2 of 4 tasks
1
[nnx] flaxlib types
#4639 opened Mar 19, 2025 by cgarciae • Draft
1
[bridge module] Add bridge.share_scope for layer-sublayer pairs.
#4638 opened Mar 19, 2025 by copybara-service bot Loading…
Add DynamicTanh (https://arxiv.org/pdf/2503.10622)
#4631 opened Mar 17, 2025 by seanemoon Loading…
1
[nnx] refactor GraphDef
#4630 opened Mar 17, 2025 by cgarciae • Draft
[nnx] remove deprecated APIs pull ready
#4627 opened Mar 15, 2025 by cgarciae Loading…
1
[nnx] add Pytree
#4625 opened Mar 15, 2025 by cgarciae Loading…
1
Implement spectral and instance norm in NNX
#4623 opened Mar 14, 2025 by mattbahr Loading…
3 of 4 tasks
1
Testing -- ignore this.
#4611 opened Mar 10, 2025 by copybara-service bot Loading…
Add pytrees as graph nodes again.
#4605 opened Mar 7, 2025 by copybara-service bot Loading…
Adding nnx Gemma2-2b (including overall fixes) to examples/gemma
#4587 opened Feb 28, 2025 by mdda Loading…
1
7
Canonicalize PartitionSpec so that we can delete ParsedPartitionSpec
#4577 opened Feb 26, 2025 by copybara-service bot Loading…
[nnx] Add NNX WeightNorm.
#4568 opened Feb 24, 2025 by nilq Loading…
3 tasks done
6
Add some methods on Sequential container
#4565 opened Feb 22, 2025 by Ruhaan838 Loading…
2 of 3 tasks
4
Fixes #1925 non-str dict keys not suppoted in module state
#4563 opened Feb 21, 2025 by muhrin Loading…
3 of 4 tasks
1
1
[nnx] deepseek example v2
#4519 opened Jan 31, 2025 by cgarciae • Draft
[nnx] deepseek example
#4512 opened Jan 29, 2025 by cgarciae • Draft
1
deepseek example
#4511 opened Jan 29, 2025 by copybara-service bot Loading…
ProTip! Filter pull requests by the default branch with base:main.