Skip to content

Pull requests: google/flax

New pull request New
100 Open 2,300 Closed
100 Open 2,300 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

[nnx] add pmap pull ready
#3969 opened Jun 5, 2024 by cgarciae Loading…
updated nnx.module docstrings
#3966 opened Jun 4, 2024 by chiamp Loading…
@chiamp
1
[linen] allow checkpoint to cache
#3965 opened Jun 4, 2024 by copybara-service bot Loading…
[nnx] experimental transforms
#3963 opened Jun 4, 2024 by cgarciae Loading…
promote nnx to root-level import pull ready
#3961 opened Jun 4, 2024 by chiamp Loading…
@chiamp
1
updated nnx.graph docstrings
#3958 opened Jun 3, 2024 by chiamp Loading…
@chiamp
5
Add more activation functions from jax.nn.
#3944 opened May 25, 2024 by carlosgmartin Loading…
Update NNX installation instructions in mnist tutorial
#3943 opened May 24, 2024 by Davidnet Loading…
1 task done
1
support custom dtype for kernel and bias parameters.
#3937 opened May 23, 2024 by copybara-service bot Loading…
1
[NNX] Make State's mapping by reference to allow attribute add/delete
#3930 opened May 22, 2024 by IvyZX Loading…
1
Support direct quantization for FP8 matmul
#3922 opened May 14, 2024 by wenscarl Loading…
24
Cleanup: add/remove apostrophes; remove backtick
#3920 opened May 13, 2024 by mbrukman Loading…
1 of 4 tasks
1
Add scale option to spectral normalization.
#3904 opened May 7, 2024 by JesseFarebro Loading…
2 of 4 tasks
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…
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…
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
ProTip! Exclude everything labeled bug with -label:bug.