Skip to content

Pull requests: google/flax

New pull request New
99 Open 2,265 Closed
99 Open 2,265 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

Replace deprecated jax.tree_* functions with jax.tree.*
#3926 opened May 15, 2024 by copybara-service bot Loading…
Added a missing return to _fingerprint_recursive
#3924 opened May 14, 2024 by copybara-service bot Loading…
upgraded haiku guide to include NNX
#3923 opened May 14, 2024 by chiamp • Draft
@chiamp
1
[draft]Support direct quantization for FP8 matmul
#3922 opened May 14, 2024 by wenscarl Loading…
9
[nnx] add lazy
#3921 opened May 13, 2024 by cgarciae Loading…
Cleanup: add/remove apostrophes; remove backtick
#3920 opened May 13, 2024 by mbrukman Loading…
1 of 4 tasks
1
[nnx] add iter_graph
#3919 opened May 10, 2024 by cgarciae Loading…
updated rng guide
#3912 opened May 8, 2024 by chiamp Loading…
@chiamp
2
[nnx] Object refactor
pull ready
#3910 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
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
remove tree_map deprecation warning
#3836 opened Apr 9, 2024 by chiamp • Draft
1
@chiamp
ProTip! Follow long discussions with comments:>50.