Skip to content

Pull requests: google/flax

New pull request New
155 Open 2,632 Closed
155 Open 2,632 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

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
Avoid returning rng states in bridge module output
#4560 opened Feb 20, 2025 by IvyZX Loading…
Add REFLECT padding to convolution layer
#4553 opened Feb 19, 2025 by sarlinpe Loading…
1 of 4 tasks
2
[nnx] pytrees are graph nodes
#4547 opened Feb 15, 2025 by cgarciae Loading…
fix trace-level detection pull ready
#4527 opened Feb 3, 2025 by cgarciae Loading…
[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…
Improve error message for when nnx.Modules use jax or numpy arrays as leaf values
#4492 opened Jan 20, 2025 by RaghuSpaceRajan Loading…
1
8
[nnx] add shard_map
#4490 opened Jan 19, 2025 by cgarciae Loading…
1
4
[nnx] improve custom_vjp ergonomics
#4489 opened Jan 19, 2025 by cgarciae Loading…
1
[linen] Linesearch (and lbfgs) support for TrainState
#4471 opened Jan 6, 2025 by emiresenov Loading…
3 of 4 tasks
8
Update NNX State.state.merge method docs in statelib.py
#4453 opened Dec 17, 2024 by 8bitmp3 Loading…
@8bitmp3
1
Update NNX State.state.filter method docs in statelib.py
#4452 opened Dec 17, 2024 by 8bitmp3 Loading…
@8bitmp3
1
Update NNX State.state.split method docs in statelib.py
#4451 opened Dec 17, 2024 by 8bitmp3 Loading…
@8bitmp3
1
Update NNX State class docs in statelib.py
#4450 opened Dec 17, 2024 by 8bitmp3 Loading…
@8bitmp3
2
Lint NNX BatchStat, Cache, Intermediate class docs in variablelib.py
#4449 opened Dec 17, 2024 by 8bitmp3 Loading…
@8bitmp3
Update NNX Variable class docs in variablelib.py
#4448 opened Dec 17, 2024 by 8bitmp3 Loading…
@8bitmp3
Lint NNX Module iter_* method docs in module.py
#4447 opened Dec 17, 2024 by 8bitmp3 Loading…
@8bitmp3
Update NNX Module eval docs in module.py
#4446 opened Dec 16, 2024 by 8bitmp3 Loading…
@8bitmp3
Update NNX Module train docs in module.py
#4445 opened Dec 16, 2024 by 8bitmp3 Loading…
@8bitmp3
Update NNX Module set_attributes docs in module.py
#4444 opened Dec 16, 2024 by 8bitmp3 Loading…
@8bitmp3
Upgrade NNX Module sow docs in module.py
#4443 opened Dec 16, 2024 by 8bitmp3 Loading…
@8bitmp3
Update NNX Module class docs in module.py
#4442 opened Dec 16, 2024 by 8bitmp3 Loading…
@8bitmp3
Update NNX state docs in graph.py
#4441 opened Dec 16, 2024 by 8bitmp3 Loading…
@8bitmp3
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.