Skip to content

Pull requests: google/flax

New pull request New
100 Open 2,254 Closed
100 Open 2,254 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Added docstring for Module.scope.path
#3913 opened May 8, 2024 by chiamp Loading…
@chiamp
1
updated rng guide
#3912 opened May 8, 2024 by chiamp Loading…
@chiamp
2
[nnx] Object refactor
#3910 opened May 8, 2024 by cgarciae Loading…
[nnx] Sequential uses regular list
#3909 opened May 8, 2024 by cgarciae Loading…
[nnx] add ConvTranspose
#3908 opened May 8, 2024 by cgarciae Loading…
added documentation to nnx.graph
#3907 opened May 7, 2024 by chiamp Loading…
@chiamp
1
parameterized NNX transforms tests
#3906 opened May 7, 2024 by chiamp Loading…
@chiamp
1
Add scale option to spectral normalization.
#3904 opened May 7, 2024 by JesseFarebro Loading…
2 of 4 tasks
[linen] test jit caching with state updates
#3900 opened May 3, 2024 by cgarciae Loading…
Add a get_inputs_kernel_bias method to the DenseGeneral layer.
#3896 opened May 1, 2024 by copybara-service bot Loading…
1
[linen] fix DenseGeneral init
#3893 opened Apr 30, 2024 by copybara-service bot Loading…
v0.8.4
#3891 opened Apr 30, 2024 by cgarciae Loading…
[nnx] fix iter_nodes
#3889 opened Apr 28, 2024 by cgarciae Loading…
3
Fix exception cause in configurations.py
#3888 opened Apr 26, 2024 by cool-RR Loading…
Let num_heads=1 by default in MultiHeadAttention.
#3883 opened Apr 24, 2024 by carlosgmartin Loading…
[nnx] add Module pytree_experimental static test
#3864 opened Apr 17, 2024 by cgarciae Loading…
1
Make ConvLSTMCell.__call__ infer the number of features from its input, like the other recurrent cells.
#3862 opened Apr 16, 2024 by carlosgmartin Loading…
Minor typo fix in flax_fundamentals/flax_basics.ipynb
#3852 opened Apr 12, 2024 by albanie Loading…
1
Minor docstring fix for flax.linen.Module
#3851 opened Apr 12, 2024 by albanie Loading…
Minor fix to docstring for flax.core.variables.py
#3850 opened Apr 12, 2024 by albanie Loading…
Minor docstring fix for flax.linen.module.variable
#3849 opened Apr 12, 2024 by albanie Loading…
Minor doc formatting fixes for flax.linen.Module.init_with_output
#3848 opened Apr 12, 2024 by albanie Loading…
1 task done
Flax NNX message in ToC
pull ready
#3845 opened Apr 11, 2024 by 8bitmp3 Loading…
@cgarciae
Fix minor typos in docstring and markdown file
#3844 opened Apr 11, 2024 by dantp-ai Loading…
1 of 4 tasks
1
remove tree_map deprecation warning
#3836 opened Apr 9, 2024 by chiamp • Draft
1
@chiamp
ProTip! Exclude everything labeled bug with -label:bug.