Skip to content

Pull requests: google/flax

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

Pull requests list

static + pytree guide
#4897 opened Aug 18, 2025 by cgarciae Loading…
1
Sharding API improvements (non breaking)
#4893 opened Aug 14, 2025 by IvyZX Loading…
3
Fix nnx.tabulate crash with empty dict/None values (fixes #4889)
#4891 opened Aug 14, 2025 by mohsinm-dev Loading…
2 of 4 tasks
1
Add python 3.13 CI tests
#4888 opened Aug 13, 2025 by IvyZX Loading…
fix_checkpoint helper
#4875 opened Aug 6, 2025 by rcrowe-google Loading…
2 tasks
Fixing formatting
#4874 opened Aug 5, 2025 by rcrowe-google Loading…
4 tasks
2
Adding link to release notes
#4873 opened Aug 5, 2025 by rcrowe-google Loading…
4 tasks
Now Github release notes
#4872 opened Aug 5, 2025 by rcrowe-google Loading…
4 tasks
Optimiser was already instantiated using the model - 05_vae.py
#4857 opened Jul 29, 2025 by nenuadrian Loading…
1 of 4 tasks
Automatically apply sharding constraints to sharded models pull ready
#4844 opened Jul 23, 2025 by IvyZX Loading…
8
Migrate rnglib.py from .value to [...]
#4840 opened Jul 23, 2025 by lukeyeh Loading…
Migrate metrics.py from .value to [...]
#4839 opened Jul 23, 2025 by lukeyeh Loading…
Migrate module_test.py from .value to [...]
#4838 opened Jul 23, 2025 by lukeyeh Loading…
Migrate containers_test.py from .value to [...]
#4837 opened Jul 23, 2025 by lukeyeh Loading…
Migrate conv_test.py from .value to [...]
#4836 opened Jul 23, 2025 by lukeyeh Loading…
Migrate embed_test.py from .value to [...]
#4835 opened Jul 23, 2025 by lukeyeh Loading…
Migrate normalization_test.py from .value to [...]
#4834 opened Jul 23, 2025 by lukeyeh Loading…
Migrate linear_test.py from .value to [...]
#4833 opened Jul 23, 2025 by lukeyeh Loading…
Migrate sampler_test.py from .value to [...]
#4832 opened Jul 23, 2025 by lukeyeh Loading…
Fixed Gemma example using Gemma2 models pull ready
#4830 opened Jul 22, 2025 by vfdev-5 Loading…
1 of 4 tasks
1
3
[nnx] update bridge
#4828 opened Jul 21, 2025 by cgarciae Loading…
Use .format in place of .layout.
#4785 opened Jun 19, 2025 by copybara-service bot Loading…
feat: Add overload signatures to module.apply
#4783 opened Jun 18, 2025 by Daraan Loading…
2 tasks done
1
Fix typo on 'Scale up on multiple devices' guide
#4765 opened May 30, 2025 by DavidUlloa6310 Loading…
@IvyZX @8bitmp3
2
[nnx] add optax optimizer
#4752 opened May 16, 2025 by cgarciae Loading…
ProTip! no:milestone will show everything without a milestone.