Skip to content

Pull requests: google/flax

New pull request New
119 Open 2,452 Closed
119 Open 2,452 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] add flaxlib
#4235 opened Sep 30, 2024 by copybara-service bot Loading…
@cgarciae
proxy more methods for nnx.Variable
#4234 opened Sep 29, 2024 by njzjz Loading…
1 of 4 tasks
Fix scale dtype and refactor q_dot_dq pull ready
#4229 opened Sep 25, 2024 by wenscarl Loading…
5
[nnx] revive TrainState toy example
#4226 opened Sep 24, 2024 by cgarciae Loading…
Utils for fp8 gated einsum
#4221 opened Sep 24, 2024 by wenscarl Loading…
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change.
#4219 opened Sep 23, 2024 by copybara-service bot Loading…
[nnx] fix mnist_tutorial link
#4204 opened Sep 17, 2024 by emmanuel-ferdman Loading…
1 of 4 tasks
1
Upgrade Flax NNX Haiku Linen migration doc
#4200 opened Sep 16, 2024 by 8bitmp3 Loading…
Upgrade Flax NNX Filters doc
#4199 opened Sep 16, 2024 by 8bitmp3 Loading…
1
Upgrade Flax NNX Bridge API doc
#4198 opened Sep 16, 2024 by 8bitmp3 Loading…
1
[WIP] Require transform_metadata when variables have sharding annotation
#4187 opened Sep 10, 2024 by IvyZX • Draft
Add miniGPT-Jax to community Examples.
#4184 opened Sep 10, 2024 by yilunyu Loading…
1 of 4 tasks
1
1
[nnx] add FSDP toy example with custom optimizer pull ready
#4183 opened Sep 9, 2024 by cgarciae Loading…
[nnx] add nnx.pytree
#4174 opened Sep 5, 2024 by cgarciae Loading…
Upgrade Flax NNX basics doc
#4173 opened Sep 5, 2024 by 8bitmp3 Loading…
1
Debug ScopeParamNotFoundError
#4163 opened Aug 31, 2024 by copybara-service bot Loading…
1
[nnx] fix ToLinen kwargs field
#4159 opened Aug 30, 2024 by copybara-service bot Loading…
3
[nnx] add Pytree
#4154 opened Aug 29, 2024 by cgarciae Loading…
6
Update imagenet.ipynb
#4128 opened Aug 18, 2024 by linsong8208 Loading…
3 tasks
1
Add an assert to check input ndim.
#4118 opened Aug 6, 2024 by copybara-service bot Loading…
1
Bump the pip group across 7 directories with 2 updates dependencies Pull requests that update a dependency file
#4111 opened Jul 30, 2024 by dependabot bot Loading…
Bump the pip group across 8 directories with 2 updates dependencies Pull requests that update a dependency file
#4110 opened Jul 30, 2024 by dependabot bot Loading…
[nnx] add call section to nnx_basics
#4090 opened Jul 18, 2024 by cgarciae Loading…
1
Fix broken link to NNX Basics in the docs
#4083 opened Jul 16, 2024 by gileshd Loading…
1 task done
[nnx] add gemma notebook
#4075 opened Jul 11, 2024 by cgarciae Loading…
3
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.