Skip to content

Pull requests: google/flax

New pull request New
154 Open 2,580 Closed
154 Open 2,580 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

[nnx] cache flatten
#4463 opened Dec 23, 2024 by cgarciae • Draft
[nnx] simplify unflatten
#4462 opened Dec 23, 2024 by cgarciae • Draft
[nnx] flatten returns FlatState
#4458 opened Dec 21, 2024 by cgarciae • Draft
[nnx] jit cache
#4455 opened Dec 19, 2024 by cgarciae Loading…
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
Lint NNX pop method docs in graph.py
#4440 opened Dec 16, 2024 by 8bitmp3 Loading…
@8bitmp3
Update NNX call method docs in graph.py
#4439 opened Dec 15, 2024 by 8bitmp3 Loading…
@8bitmp3
[nnx] add tabulate
#4438 opened Dec 15, 2024 by cgarciae • Draft
Add serialization support for np.void datatype
#4437 opened Dec 14, 2024 by copybara-service bot Loading…
1
Bypasses WSC application in flax Dense layer's unboxing before reshape.
#4435 opened Dec 13, 2024 by copybara-service bot Loading…
Fix de-serialisation of numpy fixed-width dtypes.
#4429 opened Dec 11, 2024 by copybara-service bot Loading…
1
Fixing the broken link in index.rst pull ready
#4424 opened Dec 10, 2024 by tilakrayal Loading…
4 tasks
1
Makes flax Modules more compatible with IPython auto-reload.
#4420 opened Dec 5, 2024 by copybara-service bot Loading…
1
ProTip! Exclude everything labeled bug with -label:bug.